diff --git a/scripts/ws_client.py b/scripts/ws_client.py
index 46dd2da10..233a7828a 100755
--- a/scripts/ws_client.py
+++ b/scripts/ws_client.py
@@ -234,7 +234,7 @@ async def create_client(
 
                         await protocol.on_message(ws, name, message)
 
-                    except (asyncio.TimeoutError, websockets.exceptions.WebSocketException):
+                    except (TimeoutError, websockets.exceptions.WebSocketException):
                         # Try pinging
                         try:
                             pong = await ws.ping()
@@ -244,7 +244,7 @@ async def create_client(
 
                             continue
 
-                        except asyncio.TimeoutError:
+                        except TimeoutError:
                             logger.error(f"Ping timed out, retrying in {sleep_time}s")
                             await asyncio.sleep(sleep_time)