diff --git a/claude_setup.sh b/claude_setup.sh
index 946c223..5a4e64b 100644
--- a/claude_setup.sh
+++ b/claude_setup.sh
@@ -313,11 +313,11 @@ claude_gemini() {
 
   ANTHROPIC_BASE_URL=http://localhost:8080 \
   ANTHROPIC_AUTH_TOKEN=dummy \
-  ANTHROPIC_MODEL=gemini-3.1-pro-high \
-  ANTHROPIC_DEFAULT_OPUS_MODEL=gemini-3.1-pro-high \
-  ANTHROPIC_DEFAULT_SONNET_MODEL=gemini-3.1-pro-low \
-  ANTHROPIC_DEFAULT_HAIKU_MODEL=gemini-3-flash \
-  CLAUDE_CODE_SUBAGENT_MODEL=gemini-3-flash \
+  ANTHROPIC_MODEL=gemini-pro-agent \
+  ANTHROPIC_DEFAULT_OPUS_MODEL=gemini-pro-agent \
+  ANTHROPIC_DEFAULT_SONNET_MODEL=gemini-3-flash-agent \
+  ANTHROPIC_DEFAULT_HAIKU_MODEL=gemini-3.5-flash-low \
+  CLAUDE_CODE_SUBAGENT_MODEL=gemini-3.5-flash-low \
   CLAUDE_CODE_DISABLE_NONESSENTIAL_TRAFFIC=1 \
   claude "$@"