--- a/hgext/infinitepush/__init__.py	Wed Apr 11 10:51:38 2018 -0700
+++ b/hgext/infinitepush/__init__.py	Wed Apr 11 12:49:08 2018 -0700
@@ -128,6 +128,7 @@
     util,
     wireproto,
     wireprototypes,
+    wireprotov1peer,
 )
 
 from . import (
@@ -319,7 +320,7 @@
 
     extensions.wrapfunction(discovery, 'checkheads', _checkheads)
 
-    wireproto.wirepeer.listkeyspatterns = listkeyspatterns
+    wireprotov1peer.wirepeer.listkeyspatterns = listkeyspatterns
 
     partorder = exchange.b2partsgenorder
     index = partorder.index('changeset')