hgext3rd/topic/__init__.py
changeset 5229 d8ea3c829477
parent 5008 848ca807f83a
--- a/hgext3rd/topic/__init__.py	Thu Dec 19 21:36:59 2019 -0800
+++ b/hgext3rd/topic/__init__.py	Thu Dec 26 10:21:31 2019 -0800
@@ -1282,13 +1282,11 @@
                           source=b'topic-extension')
         return orig(ui, repo, **opts)
 
-    def new_init(orig, *args, **kwargs):
-        runtime = orig(*args, **kwargs)
+    def new_init(orig, self, *args, **kwargs):
+        orig(self, *args, **kwargs)
 
-        if util.safehasattr(runtime, 'extrafns'):
-            runtime.extrafns.append(savetopic)
-
-        return runtime
+        if util.safehasattr(self, 'extrafns'):
+            self.extrafns.append(savetopic)
 
     try:
         rebase = extensions.find(b"rebase")