diff --git a/litellm/main.py b/litellm/main.py
index ed8dddf05..33fad52cc 100644
--- a/litellm/main.py
+++ b/litellm/main.py
@@ -10,7 +10,6 @@
 import os, openai, sys, json, inspect, uuid, datetime, threading
 from typing import Any, Literal, Union
 from functools import partial
-
 import dotenv, traceback, random, asyncio, time, contextvars
 from copy import deepcopy
 import httpx
diff --git a/litellm/tests/test_amazing_vertex_completion.py b/litellm/tests/test_amazing_vertex_completion.py
index bca241cd7..35beb75fe 100644
--- a/litellm/tests/test_amazing_vertex_completion.py
+++ b/litellm/tests/test_amazing_vertex_completion.py
@@ -250,6 +250,8 @@ async def test_async_vertexai_streaming_response():
             "code-gecko@latest",
             "code-bison@001",
             "text-bison@001",
+            "gemini-1.5-pro",
+            "gemini-1.5-pro-vision",
         ]:
             # our account does not have access to this model
             continue