--- a/tests/test-sshserver.py	Wed Jan 31 11:13:11 2018 -0800
+++ b/tests/test-sshserver.py	Wed Jan 31 10:48:35 2018 -0800
@@ -6,9 +6,9 @@
 import silenttestrunner
 
 from mercurial import (
-    sshserver,
     util,
     wireproto,
+    wireprotoserver,
 )
 
 class SSHServerGetArgsTests(unittest.TestCase):
@@ -29,7 +29,7 @@
 def mockserver(inbytes):
     ui = mockui(inbytes)
     repo = mockrepo(ui)
-    return sshserver.sshserver(ui, repo)
+    return wireprotoserver.sshserver(ui, repo)
 
 class mockrepo(object):
     def __init__(self, ui):