--- a/tests/test-batching.py	Wed Apr 11 10:51:38 2018 -0700
+++ b/tests/test-batching.py	Wed Apr 11 12:49:08 2018 -0700
@@ -11,7 +11,7 @@
     error,
     peer,
     util,
-    wireproto,
+    wireprotov1peer,
 )
 
 # equivalent of repo.repository
@@ -177,7 +177,7 @@
             yield r
 
     def batchiter(self):
-        return wireproto.remoteiterbatcher(self)
+        return wireprotov1peer.remoteiterbatcher(self)
 
     @peer.batchable
     def foo(self, one, two=None):