revert to base_class

Drew-Goddyn · Drew-Goddyn · commit 86fa07084ad5 · 2020-11-18T15:34:45.000-08:00
diff --git a/lib/polymorphic_integer_type/belongs_to_polymorphic_association_extension.rb b/lib/polymorphic_integer_type/belongs_to_polymorphic_association_extension.rb
@@ -3,7 +3,7 @@ module Associations
     class BelongsToPolymorphicAssociation < BelongsToAssociation
       private def replace_keys(record)
         super
-        owner[reflection.foreign_type] = record.class
+        owner[reflection.foreign_type] = record.class.base_class
       end
     end
   end
