diff --git a/tests/unit_tests/test_subtensor.py b/tests/unit_tests/test_subtensor.py
index 73c63b30b9..740df91884 100644
--- a/tests/unit_tests/test_subtensor.py
+++ b/tests/unit_tests/test_subtensor.py
@@ -256,6 +256,9 @@ def subtensor(mocker):
     mocker.patch.object(
         subtensor_module, "SubstrateInterface", return_value=fake_substrate
     )
+    fake_websocket = mocker.MagicMock()
+    fake_websocket.client.connect.return_value = 0
+    mocker.patch.object(subtensor_module, "ws_client", return_value=fake_websocket)
     return Subtensor()