diff --git a/src/services/chatbot.py b/src/services/chatbot.py
index 4b66d0f..719a803 100644
--- a/src/services/chatbot.py
+++ b/src/services/chatbot.py
@@ -459,7 +459,7 @@ async def send_message(
         from .kiaai import KiaAIService
 
         # Push the human message into the graph state first
-        await app.aupdate_state(config, {"messages": [human_msg]})
+        await app.aupdate_state(config, {"messages": [human_msg]}, as_node="model")
         state = await app.aget_state(config)
         all_msgs = state.values.get("messages", [human_msg])
         trimmed = await get_trimmer().ainvoke(all_msgs)
@@ -506,7 +506,7 @@ async def send_message(
             "total_tokens": input_tokens + output_tokens,
         }
         ai_msg.additional_kwargs["created_at"] = ai_msg_created_at
-        await app.aupdate_state(config, {"messages": [ai_msg]})
+        await app.aupdate_state(config, {"messages": [ai_msg]}, as_node="model")
 
         await user.save()
         await Message.create(