remove redundant identity map set

1 files changed, 2 insertions, 1 deletions

diff --git a/lib/sqlalchemy/orm/unitofwork.py b/lib/sqlalchemy/orm/unitofwork.py
index 02c230d08..19642cc9d 100644
--- a/lib/sqlalchemy/orm/unitofwork.py
+++ b/lib/sqlalchemy/orm/unitofwork.py
@@ -122,14 +122,15 @@ class UnitOfWork(object):
             
         elif state.dict['_instance_key'] != instance_key:
             # primary key switch
-            self.identity_map[instance_key] = state.obj()
             del self.identity_map[state.dict['_instance_key']]
             state.dict['_instance_key'] = instance_key
             
         if hasattr(state, 'insert_order'):
             delattr(state, 'insert_order')
+            
         self.identity_map[state.dict['_instance_key']] = state.obj()
         state.commit_all()
+        
         # remove from new last, might be the last strong ref
         self.new.pop(state, None)