diff --git a/src/bonsai/bonsai/bim/module/tester/operator.py b/src/bonsai/bonsai/bim/module/tester/operator.py
index 63906ba25d..767c223985 100644
--- a/src/bonsai/bonsai/bim/module/tester/operator.py
+++ b/src/bonsai/bonsai/bim/module/tester/operator.py
@@ -106,7 +106,7 @@ class IfcTesterWebSocketServer:
                 self.runner = None
 
             print("IfcTester WebSocket server stopped")
-        except asyncio.TimeoutError:
+        except TimeoutError:
             print("Websocket server cleanup timed out, forcing shutdown")
         except Exception as e:
             print(f"Error during websocket cleanup: {e}")