diff -r fd9f50406cb8 -r 6aa349f47e7c states.py
--- a/states.py	Wed May 25 02:26:38 2011 +0200
+++ b/states.py	Wed May 25 02:27:09 2011 +0200
@@ -124,8 +124,6 @@
     }
 
 
-templatekw.keywords['state'] = showstate
-
 
 
 
@@ -156,6 +154,8 @@
     wireproto.wirerepository._reducehead = _reducehead
     wireproto.commands['heads'] = (heads, '')
 
+    templatekw.keywords['state'] = showstate
+
 def extsetup(ui):
     for state in STATES:
         if state.trackheads: