diff --git a/pyhilo/api.py b/pyhilo/api.py
index 84f7214..1c9cc2c 100755
--- a/pyhilo/api.py
+++ b/pyhilo/api.py
@@ -373,8 +373,8 @@ async def _async_post_init(self) -> None:
         await self.websocket_manager.initialize_websockets()
 
         # Create both websocket clients
-        self.websocket = WebsocketClient(self, self.websocket_manager.devicehub)
-        self.websocket2 = WebsocketClient(self, self.websocket_manager.challengehub)
+        self.websocket = WebsocketClient(self.websocket_manager.devicehub)
+        self.websocket2 = WebsocketClient(self.websocket_manager.challengehub)
 
     async def refresh_ws_token(self) -> None:
         """Refresh the websocket token."""