--- 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,