diff --git a/mm_agents/agent.py b/mm_agents/agent.py
index d947fb4..42aa53e 100644
--- a/mm_agents/agent.py
+++ b/mm_agents/agent.py
@@ -837,7 +837,7 @@ class PromptAgent:
             )
             return response.text
 
-        elif self.model == "gemini-1.5-pro-latest":
+        elif self.model.startswith("gemini"):
             messages = payload["messages"]
             max_tokens = payload["max_tokens"]
             top_p = payload["top_p"]