diff --git a/utils/agent.py b/utils/agent.py
index 5ae9b61..57b5ff7 100644
--- a/utils/agent.py
+++ b/utils/agent.py
@@ -13,7 +13,7 @@ def create_clevrr_agent(model, prompt):
     df_locals = {}
     df_locals["pg"] = pg
     tools = [PythonAstREPLTool(locals=df_locals), get_screen_info]
-    model = MODELS["openai"]
+    model = MODELS["gemini"]
     agent = create_react_agent(model, tools, prompt)
     agent_executor = AgentExecutor(agent=agent, tools=tools, verbose=True, handle_parsing_errors=True, return_intermediate_steps=True)
-    return agent_executor
\ No newline at end of file
+    return agent_executor