diff --git a/README.md b/README.md
index f50bf8a..a51fd85 100644
--- a/README.md
+++ b/README.md
@@ -160,7 +160,7 @@ import random
 import os
 
 base_url = "/api/v1/gateway/litellm/{add-your-dataset-name-here}"
-EXAMPLE_MODELS = ["openai/gpt-4o", "gemini/gemini-2.0-flash", "anthropic/claude-3-5-haiku-20241022"]
+EXAMPLE_MODELS = ["openai/gpt-4o", "gemini/gemini-2.5-flash", "anthropic/claude-3-5-haiku-20241022"]
 model = random.choice(SAMPLE_MODELS)
 
 base_url += "/" + model.split("/")[0] # append /gemini /openai or /anthropic. 
diff --git a/tests/integration/litellm/test_chat_without_tool_call.py b/tests/integration/litellm/test_chat_without_tool_call.py
index ee831a9..04b2880 100644
--- a/tests/integration/litellm/test_chat_without_tool_call.py
+++ b/tests/integration/litellm/test_chat_without_tool_call.py
@@ -10,7 +10,7 @@ from litellm import completion
 
 MODEL_API_KEYS = {
     "openai/gpt-4o": "OPENAI_API_KEY",
-    "gemini/gemini-2.0-flash": "GEMINI_API_KEY",
+    "gemini/gemini-2.5-flash": "GEMINI_API_KEY",
     "anthropic/claude-3-5-haiku-20241022": "ANTHROPIC_API_KEY",
 }