diff --git a/ui/components/ChatWindow.tsx b/ui/components/ChatWindow.tsx
index ea9a93d..34e4114 100644
--- a/ui/components/ChatWindow.tsx
+++ b/ui/components/ChatWindow.tsx
@@ -194,7 +194,7 @@ const useSocket = (
 
         ws.onclose = () => {
           clearTimeout(timeoutId);
-          setError(true);
+          setWs(null);
           console.log('[DEBUG] closed');
         };