lib/memento/action/destroy.rb in memento-0.3.6 vs lib/memento/action/destroy.rb in memento-0.3.7

@@ -10,10 +10,11 @@
         object.save!
       rescue
         object.id = nil
         object.save!
       end
-      state.update_attribute(:record, object)
+      state.record = object
+      state.save
     end
   end
   
   private
   
\ No newline at end of file