diff --git a/src/bin/bind10/tests/bind10_test.py.in b/src/bin/bind10/tests/bind10_test.py.in
index f6ad0e9..1eb2fcb 100644
--- a/src/bin/bind10/tests/bind10_test.py.in
+++ b/src/bin/bind10/tests/bind10_test.py.in
@@ -429,10 +429,14 @@ class TestBoB(unittest.TestCase):
 
         # Fake the _get_socket, which is complicated and tested elsewhere
         # We just want to pass the parameters in and let it create a response
-        def get_socket(args):
-            return isc.config.ccsession.create_answer(0, args)
-
-        bob._get_socket = get_socket
+#         def get_socket(args):
+#             return isc.config.ccsession.create_answer(0, args)
+
+#         bob._get_socket = get_socket
+        def get_token(protocol, addr, port, share_mode, share_name):
+            return str(addr) + ':' + str(port)
+        bob._socket_cache = bob
+        bob._socket_cache.get_token = get_token
         args = {
             "port": 53,
             "address": "0.0.0.0",
@@ -443,7 +447,8 @@ class TestBoB(unittest.TestCase):
         # Test it just returns whatever it got. The real function doesn't
         # work like this, but we don't want the command_handler to touch it
         # at all and this is the easiest way to check.
-        self.assertEqual({'result': [0, args]},
+        self.assertEqual({'result': [0, {'token': '0.0.0.0:53',
+                                         'path': None}]},
                          bob.command_handler("get_socket", args))
         # The drop_socket is not tested here, but in TestCacheCommands.
         # It needs the cache mocks to be in place and they are there.