diff --git a/src/server/deep_agent/agents/main_agent.py b/src/server/deep_agent/agents/main_agent.py
index 4c28afd..177c85e 100755
--- a/src/server/deep_agent/agents/main_agent.py
+++ b/src/server/deep_agent/agents/main_agent.py
@@ -141,11 +141,11 @@ def build_main_agent(workspace_dir, enable_thinking):
     )
     main_agent = create_deep_agent(
         model=build_main_llm(enable_thinking=enable_thinking),
-        # store=InMemoryStore(),
+        store=InMemoryStore(),
         subagents=subagents,
-        # checkpointer=checkpointer,
+        checkpointer=checkpointer,
         tools=[edit_furniture, generate_furniture, edit_quote_upload_furniture],
-        # backend=backend,
+        backend=backend,
         context_schema=Context,
         middleware=middleware,
     )