--- a/states.py	Mon Jun 06 16:42:56 2011 +0200
+++ b/states.py	Fri Jul 01 14:55:02 2011 +0200
@@ -193,8 +193,9 @@
         return
 
     ocancopy =repo.cancopy
-    opull =repo.pull
-    opush =repo.push
+    opull = repo.pull
+    opush = repo.push
+    o_tag = repo._tag
     class statefulrepo(repo.__class__):
 
         def nodestate(self, node):
@@ -332,5 +333,10 @@
                 remote = map(node.bin, remote.listkeys('immutableheads'))
             return remote
 
+        def _tag(self, names, node, *args, **kwargs):
+            tagnode = o_tag(names, node, *args, **kwargs)
+            self.setstate(ST0, [node, tagnode])
+            return tagnode
+
     repo.__class__ = statefulrepo