diff --git a/src/backend/langflow/api/chat_manager.py b/src/backend/langflow/api/chat_manager.py
index 6a4111fb4..7ada67793 100644
--- a/src/backend/langflow/api/chat_manager.py
+++ b/src/backend/langflow/api/chat_manager.py
@@ -148,7 +148,7 @@ class ChatManager:
             files=file_responses,
         )
         await self.send_json(client_id, response)
-        self.chat_history.add_message(client_id, chat_message)
+        self.chat_history.add_message(client_id, response)
 
     async def handle_websocket(self, client_id: str, websocket: WebSocket):
         await self.connect(client_id, websocket)