diff --git a/src/components/GenerateGuide/AIGuideChat.tsx b/src/components/GenerateGuide/AIGuideChat.tsx
index 401dac35f..4081937ff 100644
--- a/src/components/GenerateGuide/AIGuideChat.tsx
+++ b/src/components/GenerateGuide/AIGuideChat.tsx
@@ -297,14 +297,13 @@ export function AIGuideChat(props: AIGuideChatProps) {
 
             <button
               className="absolute top-1/2 right-2 -translate-y-1/2 p-1 text-zinc-500 hover:text-black disabled:opacity-50"
-              disabled={isStreamingMessage}
               onClick={() => {
                 if (!isLoggedIn()) {
                   showLoginPopup();
                   return;
                 }
 
-                if (isStreamingMessage) {
+                if (status !== 'idle') {
                   stop();
                   return;
                 }
diff --git a/src/hooks/use-chat.ts b/src/hooks/use-chat.ts
index 48eedb666..833fa74ed 100644
--- a/src/hooks/use-chat.ts
+++ b/src/hooks/use-chat.ts
@@ -100,6 +100,7 @@ export function useChat(options: UseChatOptions) {
           },
         });
 
+        setStatus('idle');
         abortControllerRef.current = null;
         onFinish?.();
       } catch (error) {
@@ -110,6 +111,7 @@ export function useChat(options: UseChatOptions) {
         }
 
         onError?.(error as Error);
+        setStatus('error');
       }
     },
     [endpoint, onError],