diff --git a/moonraker/websockets.py b/moonraker/websockets.py
index 5163f81..feeb6c4 100644
--- a/moonraker/websockets.py
+++ b/moonraker/websockets.py
@@ -372,7 +372,7 @@ class WebsocketManager(APITransport):
         for ws in list(self.websockets.values()):
             ws.close()
         try:
-            await self.closed_event.wait(2.)
+            await self.closed_event.wait(IOLoop.current().time() + 2.)
         except tornado.util.TimeoutError:
             pass
         self.closed_event = None