diff --git a/lang_agent/graphs/routing.py b/lang_agent/graphs/routing.py
index dd6470a..f70d865 100644
--- a/lang_agent/graphs/routing.py
+++ b/lang_agent/graphs/routing.py
@@ -166,7 +166,7 @@ class RoutingGraph(GraphBase):
 
 
         out = self.chat_model.invoke(*inp)
-        return {"messages": out}
+        return {"messages": out["messages"]}
 
 
     def _tool_model_call(self, state:State):