diff --git a/src/agent.py b/src/agent.py
index 47aca58..be0f1f1 100644
--- a/src/agent.py
+++ b/src/agent.py
@@ -121,7 +121,6 @@ TOOLS AVAILABLE:
 - Restart a pod or deployment: use the `delete_k8s_pod` tool.
 """
     
-    # Use state_modifier or prompt depending on langgraph version, falling back to older syntax if needed.
-    agent = create_react_agent(llm, tools, messages_modifier=system_prompt)
-    result = agent.invoke({"messages": [("user", user_request)]})
+    agent = create_react_agent(llm, tools)
+    result = agent.invoke({"messages": [("system", system_prompt), ("user", user_request)]})
     return result["messages"][-1].content