diff --git a/litellm/tests/test_google_ai_studio_gemini.py b/litellm/tests/test_google_ai_studio_gemini.py
index 7cebd25372..5012717d38 100644
--- a/litellm/tests/test_google_ai_studio_gemini.py
+++ b/litellm/tests/test_google_ai_studio_gemini.py
@@ -6,29 +6,34 @@ sys.path.insert(
 import litellm
 from dotenv import load_dotenv
 
+
 def generate_text():
     try:
+        litellm.set_verbose = True
         messages = [
             {
                 "role": "user",
                 "content": [
-                    {
-                        "type": "text",
-                        "text": "What is this image?"
-                    },
+                    {"type": "text", "text": "What is this image?"},
                     {
                         "type": "image_url",
                         "image_url": {
                             "url": "https://avatars.githubusercontent.com/u/17561003?v=4"
-                        }
-                    }
-                ]
+                        },
+                    },
+                ],
             }
         ]
-        response = litellm.completion(model="gemini/gemini-pro-vision", messages=messages, stop="Hello world")
+        response = litellm.completion(
+            model="gemini/gemini-pro-vision",
+            messages=messages,
+            stop="Hello world",
+            num_retries=3,
+        )
         print(response)
         assert isinstance(response.choices[0].message.content, str) == True
     except Exception as exception:
         raise Exception("An error occurred during text generation:", exception)
 
-generate_text()
+
+# generate_text()