diff --git a/examples/chat-ui/src/components/ReasoningBlock.tsx b/examples/chat-ui/src/components/ReasoningBlock.tsx
index c8b6d7f..7b21fa1 100644
--- a/examples/chat-ui/src/components/ReasoningBlock.tsx
+++ b/examples/chat-ui/src/components/ReasoningBlock.tsx
@@ -15,14 +15,10 @@ const ReasoningBlock: React.FC<ReasoningBlockProps> = ({
 }) => {
     const [isExpanded, setIsExpanded] = useState(isStreaming); // Start expanded when streaming
     
-    // Auto-collapse when streaming finishes
+    // Auto-collapse instantly when streaming finishes
     useEffect(() => {
         if (!isStreaming && startTime && endTime) {
-            // Add a small delay before collapsing so user can see completion
-            const timer = setTimeout(() => {
-                setIsExpanded(false);
-            }, 500);
-            return () => clearTimeout(timer);
+            setIsExpanded(false);
         }
     }, [isStreaming, startTime, endTime]);