diff --git a/components/multimodal-input.tsx b/components/multimodal-input.tsx
index 19b281051e..f37510c0ba 100644
--- a/components/multimodal-input.tsx
+++ b/components/multimodal-input.tsx
@@ -253,7 +253,7 @@ function PureMultimodalInput({
       </div>
 
       <div className="absolute bottom-0 right-0 p-2 w-fit flex flex-row justify-end">
-        {status === 'submitted' ? (
+        {['submitted', 'streaming'].includes(status) ? (
           <StopButton stop={stop} setMessages={setMessages} />
         ) : (
           <SendButton