switch to safe operators

Drew-Goddyn · Drew-Goddyn · commit afeac5aece8d · 2020-11-20T14:09:56.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,9 +3,7 @@ module Associations
     class BelongsToPolymorphicAssociation < BelongsToAssociation
       private def replace_keys(record)
         super
-        klass = record.nil? ? record.class : record.class.base_class
-
-        owner[reflection.foreign_type] = klass
+        owner[reflection.foreign_type] = record&.class&.base_class
       end
     end
   end
