diff --git a/examples/mcp_example/math_server.py b/examples/mcp_example/math_server.py
index ca7eaf2e..4f3b111b 100644
--- a/examples/mcp_example/math_server.py
+++ b/examples/mcp_example/math_server.py
@@ -36,4 +36,6 @@ def calculate_percentage(part: float, whole: float) -> float:
 
 if __name__ == "__main__":
     print("Starting Math Server on port 6274...")
-    mcp.run(transport="sse", host="0.0.0.0", port=6274)
+    # Initialize LiteLLM with specific model
+llm = LiteLLM(model_name="gpt-4o-mini")
+mcp.run(transport="sse", host="0.0.0.0", port=6274)
diff --git a/examples/mcp_example/test_integration.py b/examples/mcp_example/test_integration.py
index 0a521499..edce69db 100644
--- a/examples/mcp_example/test_integration.py
+++ b/examples/mcp_example/test_integration.py
@@ -21,7 +21,8 @@ def setup_agent(name: str, description: str, servers: list) -> Agent:
         system_prompt="You are a math assistant. Process mathematical operations using the provided MCP tools.",
         max_loops=1,
         mcp_servers=servers,
-        streaming_on=False
+        streaming_on=False,
+        model_name="gpt-4o-mini" # Added model_name here
     )
 
 def main():