diff --git a/tests/local_testing/test_amazing_vertex_completion.py b/tests/local_testing/test_amazing_vertex_completion.py
index 9782bf3c2af..7f13aaf67c6 100644
--- a/tests/local_testing/test_amazing_vertex_completion.py
+++ b/tests/local_testing/test_amazing_vertex_completion.py
@@ -4223,7 +4223,7 @@ def test_gemini_google_maps_tool_simple():
             )
         print(f"Response: {response.model_dump_json(indent=4)}")
         assert response.choices[0].message.content is not None
-    except litellm.RateLimitError:
+    except (litellm.RateLimitError, litellm.InternalServerError):
         pass
     except Exception as e:
         pytest.fail(f"Error occurred: {e}")