diff --git a/apps/web/src/layouts/ChatLayout/ChatContainer/ChatContent/MessageContainer.tsx b/apps/web/src/layouts/ChatLayout/ChatContainer/ChatContent/MessageContainer.tsx
index fdfa9ecd1..09c9f3951 100644
--- a/apps/web/src/layouts/ChatLayout/ChatContainer/ChatContent/MessageContainer.tsx
+++ b/apps/web/src/layouts/ChatLayout/ChatContainer/ChatContent/MessageContainer.tsx
@@ -47,7 +47,7 @@ export const MessageContainer = forwardRef<HTMLDivElement, MessageContainerProps
           ) : (
             <BusterLoadingAvatar
               loading={!isStreamFinished && !isFinishedReasoning}
-              variant={hasReasoningMessage ? 'default' : 'gray'}
+              variant={hasReasoningMessage || isStreamFinished ? 'default' : 'gray'}
             />
           )}
         </div>