diff --git a/lang_agent/graphs/routing.py b/lang_agent/graphs/routing.py
index 1d34e8e..7b24b8c 100644
--- a/lang_agent/graphs/routing.py
+++ b/lang_agent/graphs/routing.py
@@ -77,8 +77,8 @@ class RoutingGraph(GraphBase):
         self.router = self.llm.with_structured_output(Route)
 
         tool_manager:ToolManager = self.config.tool_manager_config.setup()
-        self.chat_model = create_agent(self.llm, [], self.memory)
-        self.tool_model = create_agent(self.llm, tool_manager.get_langchain_tools(), self.memory)
+        self.chat_model = create_agent(self.llm, [], checkpointer=self.memory)
+        self.tool_model = create_agent(self.llm, tool_manager.get_langchain_tools(), checkpointer=self.memory)
 
 
     def _router_call(self, state:State):