--- a/tests/test-sshserver.py	Wed Feb 14 17:35:13 2018 -0700
+++ b/tests/test-sshserver.py	Thu Feb 08 15:09:59 2018 -0800
@@ -23,8 +23,11 @@
 
     def assertparse(self, cmd, input, expected):
         server = mockserver(input)
+        proto = wireprotoserver.sshv1protocolhandler(server._ui,
+                                                     server._fin,
+                                                     server._fout)
         _func, spec = wireproto.commands[cmd]
-        self.assertEqual(server._proto.getargs(spec), expected)
+        self.assertEqual(proto.getargs(spec), expected)
 
 def mockserver(inbytes):
     ui = mockui(inbytes)