diff --git a/tests/test_litellm/llms/gemini/realtime/test_gemini_realtime_transformation.py b/tests/test_litellm/llms/gemini/realtime/test_gemini_realtime_transformation.py
index d175de7dc47..4806a7e2737 100644
--- a/tests/test_litellm/llms/gemini/realtime/test_gemini_realtime_transformation.py
+++ b/tests/test_litellm/llms/gemini/realtime/test_gemini_realtime_transformation.py
@@ -102,8 +102,10 @@ def test_gemini_realtime_transformation_content_delta():
     assert config is not None
 
     session_configuration_request = {
-        "model": "gemini-1.5-flash",
-        "generationConfig": {"responseModalities": ["TEXT"]},
+        "setup": {
+            "model": "gemini-1.5-flash",
+            "generationConfig": {"responseModalities": ["TEXT"]},
+        }
     }
     session_configuration_request_str = json.dumps(session_configuration_request)
     session_created_message = {
@@ -195,8 +197,10 @@ def test_gemini_realtime_transformation_audio_delta():
     assert config is not None
 
     session_configuration_request = {
-        "model": "gemini-1.5-flash",
-        "generationConfig": {"responseModalities": ["AUDIO"]},
+        "setup": {
+            "model": "gemini-1.5-flash",
+            "generationConfig": {"responseModalities": ["AUDIO"]},
+        }
     }
     session_configuration_request_str = json.dumps(session_configuration_request)
 
@@ -244,8 +248,10 @@ def test_gemini_realtime_transformation_generation_complete():
     assert config is not None
 
     session_configuration_request = {
-        "model": "gemini-1.5-flash",
-        "generationConfig": {"responseModalities": ["AUDIO"]},
+        "setup": {
+            "model": "gemini-1.5-flash",
+            "generationConfig": {"responseModalities": ["AUDIO"]},
+        }
     }
     session_configuration_request_str = json.dumps(session_configuration_request)