diff -r 9a745ced79a9 -r c256ae48fd26 hgext/record.py
--- a/hgext/record.py	Tue Mar 10 14:42:07 2015 -0700
+++ b/hgext/record.py	Tue Mar 10 17:03:40 2015 -0700
@@ -253,7 +253,11 @@
         for chunk, l in patch.difflabel(lambda: args):
             orig(chunk, label=label + l)
     oldwrite = ui.write
-    extensions.wrapfunction(ui, 'write', wrapwrite)
+
+    def wrap(*args, **kwargs):
+        return wrapwrite(oldwrite, *args, **kwargs)
+    setattr(ui, 'write', wrap)
+
     try:
         return cmdutil.commit(ui, repo, recordfunc, pats, opts)
     finally: