1 files changed, 8 insertions, 0 deletions

diff --git a/lib/sqlalchemy/objectstore.py b/lib/sqlalchemy/objectstore.py
index 3525e40b4..db1631d39 100644
--- a/lib/sqlalchemy/objectstore.py
+++ b/lib/sqlalchemy/objectstore.py
@@ -135,6 +135,14 @@ class UnitOfWork(object):
         if hasattr(obj, "_instance_key"):
             del self.identity_map[obj._instance_key]
         del self.deleted[obj]
+        try:
+            del self.dirty[obj]
+        except KeyError:
+            pass
+        try:
+            del self.new[obj]
+        except KeyError:
+            pass
         self.attributes.remove(obj)
         
     def update(self, obj):