diff --git a/hitl_agent.py b/hitl_agent.py
index 63e126a..bf99cbb 100644
--- a/hitl_agent.py
+++ b/hitl_agent.py
@@ -3,6 +3,7 @@ from __future__ import annotations
 from typing import TypedDict
 
 from langgraph.graph import END, START, StateGraph
+from langgraph.checkpoint.memory import InMemorySaver
 from langgraph.types import Command, interrupt
 
 
@@ -50,7 +51,7 @@ def build_graph():
     graph.add_edge("intro", "ask_human")
     graph.add_edge("ask_human", "resolve")
     graph.add_edge("resolve", END)
-    return graph.compile()
+    return graph.compile(checkpointer=InMemorySaver())
 
 
 def main() -> None: