diff --git a/packages/frontend/core/src/blocksuite/ai/components/ai-chat-input/ai-chat-input.ts b/packages/frontend/core/src/blocksuite/ai/components/ai-chat-input/ai-chat-input.ts
index af2e1dbf2..d4a3a8e0b 100644
--- a/packages/frontend/core/src/blocksuite/ai/components/ai-chat-input/ai-chat-input.ts
+++ b/packages/frontend/core/src/blocksuite/ai/components/ai-chat-input/ai-chat-input.ts
@@ -426,7 +426,7 @@ export class AIChatInput extends SignalWatcher(WithDisposable(LitElement)) {
             ? html`<span class="chat-input-icon-label">Reason</span>`
             : nothing}
         </div>
-        ${status === 'transmitting'
+        ${status === 'transmitting' || status === 'loading'
           ? html`<div @click=${this._handleAbort} data-testid="chat-panel-stop">
               ${ChatAbortIcon}
             </div>`