diff --git a/src/components/thread/index.tsx b/src/components/thread/index.tsx
index d3368da..0e56068 100644
--- a/src/components/thread/index.tsx
+++ b/src/components/thread/index.tsx
@@ -74,8 +74,9 @@ export function Thread() {
       messages[messages.length - 1].type === "ai"
     ) {
       setFirstTokenReceived(true);
-      prevMessageLength.current = messages.length;
     }
+
+    prevMessageLength.current = messages.length;
   }, [messages]);
 
   const handleSubmit = (e: FormEvent) => {