lib/mongo_mapper/plugins/dirty.rb in jmonteiro-mongo_mapper-0.1.6 vs lib/mongo_mapper/plugins/dirty.rb in jmonteiro-mongo_mapper-0.1.7

@@ -36,12 +36,12 @@
 
         def changes
           changed.inject({}) { |h, key| h[key] = key_change(key); h }
         end
 
-        def initialize(attrs={})
+        def initialize(*args)
           super
-          changed_keys.clear if attrs.blank?
+          changed_keys.clear if args.first.blank? || !new?
         end
 
         def save(*args)
           if status = super
             changed_keys.clear
\ No newline at end of file