diff --git a/tests/tests_devices/test_stddaq_client.py b/tests/tests_devices/test_stddaq_client.py
index 75e777f..c4fc124 100644
--- a/tests/tests_devices/test_stddaq_client.py
+++ b/tests/tests_devices/test_stddaq_client.py
@@ -8,18 +8,14 @@ import typeguard
 from ophyd import StatusBase
 from websockets import WebSocketException
 
-from tomcat_bec.devices.gigafrost.std_daq_client import (
-    StdDaqClient,
-    StdDaqError,
-    StdDaqStatus,
-)
+from tomcat_bec.devices.gigafrost.std_daq_client import StdDaqClient, StdDaqError, StdDaqStatus
 
 
 @pytest.fixture
 def client():
     parent_device = mock.MagicMock()
     _client = StdDaqClient(
-        parent=parent_device, ws_url="http://localhost:5000", rest_url="http://localhost:5000"
+        parent=parent_device, ws_url="ws://localhost:5000", rest_url="http://localhost:5000"
     )
     yield _client
     _client.shutdown()