diff -r 272a8a316f7a -r e01f1b4a62e9 hgkw/keyword.py
--- a/hgkw/keyword.py	Sun Jan 23 03:15:39 2011 +0100
+++ b/hgkw/keyword.py	Sun Jan 23 03:15:44 2011 +0100
@@ -636,8 +636,6 @@
         finally:
             wlock.release()
 
-    repo.__class__ = kwrepo
-
     def kwfilectx_cmp(orig, self, fctx):
         # keyword affects data size, comparing wdir and filelog size does
         # not make sense
@@ -661,6 +659,8 @@
         except KeyError:
             pass
 
+    repo.__class__ = kwrepo
+
 cmdtable = {
     'kwdemo':
         (demo,