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