Should always try to do the lookup unless it is an integer

Author: Kyle d'Oliveira

lib/polymorphic_integer_type/extensions.rb

@@ -17,10 +17,7 @@ def belongs_to(name, options = {})
           end
 
           define_method "#{foreign_type}=" do |klass|
-            case klass
-            when Integer then enum = klass
-            when String then enum = mapping.key(klass)
-            end
+            enum = mapping.key(klass.to_s) || klass
             super(enum)
           end