--- a/hgext/infinitepush/__init__.py	Wed Apr 11 12:49:08 2018 -0700
+++ b/hgext/infinitepush/__init__.py	Wed Apr 11 12:51:09 2018 -0700
@@ -120,7 +120,6 @@
     extensions,
     hg,
     localrepo,
-    peer,
     phases,
     pushkey,
     pycompat,
@@ -354,11 +353,11 @@
     else:
         return orig(self, namespace)
 
-@peer.batchable
+@wireprotov1peer.batchable
 def listkeyspatterns(self, namespace, patterns):
     if not self.capable('pushkey'):
         yield {}, None
-    f = peer.future()
+    f = wireprotov1peer.future()
     self.ui.debug('preparing listkeys for "%s" with pattern "%s"\n' %
                   (namespace, patterns))
     yield {