force cancelling the task

kraken/futures/websocket/__init__.py

@@ -86,7 +86,12 @@ async def stop(self: ConnectFuturesWebsocket) -> None:
         """Stops the websocket connection"""
         self.keep_alive = False
         if hasattr(self, "task") and not self.task.done():
-            await self.task
+            await self.socket.close()
+            try:
+                self.task.cancel()
+            except asyncio.CancelledError:
+                # Task was successfully cancelled
+                pass
 
     async def __run(  # noqa: C901
         self: ConnectFuturesWebsocket,
@@ -97,6 +102,7 @@ async def __run(  # noqa: C901
 
         async with websockets.connect(  # pylint: disable=no-member # noqa: PLR1702
             f"wss://{self.__ws_endpoint}",
+            dditional_headers={"User-Agent": "btschwertfeger/python-kraken-sdk"},
             ping_interval=30,
         ) as socket:
             LOG.info("Websocket connected!")

kraken/spot/websocket/connectors.py

@@ -114,7 +114,6 @@ async def stop(self: ConnectSpotWebsocketBase) -> None:
             await self.socket.close()
             try:
                 self.task.cancel()
-                await self.task  # Await to handle any cancellation cleanup
             except asyncio.CancelledError:
                 # Task was successfully cancelled
                 pass