diff --git a/lib/chat/actions.tsx b/lib/chat/actions.tsx
index a97e0676..f5b69205 100644
--- a/lib/chat/actions.tsx
+++ b/lib/chat/actions.tsx
@@ -276,7 +276,7 @@ async function submitUserMessage(content: string) {
           textContent += textDelta
           messageStream.update(<BotMessage content={textContent} />)
 
-          aiState.update({
+          aiState.done({
             ...aiState.get(),
             messages: [
               ...aiState.get().messages,