lib/evil_seed/anonymizer.rb in evil-seed-0.1.3 vs lib/evil_seed/anonymizer.rb in evil-seed-0.2.0

@@ -31,16 +31,16 @@
         end
       end
     end
 
     def respond_to_missing?(attribute_name)
-      @model_class.attribute_names.include?(attribute_name.to_s) || super
+      @model_class.column_names.include?(attribute_name.to_s) || super
     end
 
     private
 
     def method_missing(attribute_name, &block)
-      return super unless @model_class.attribute_names.include?(attribute_name.to_s)
+      return super unless @model_class.column_names.include?(attribute_name.to_s)
       @changers[attribute_name.to_s] = block
     end
   end
 end