diff --git a/litellm/tests/test_secrets.py b/litellm/tests/test_secrets.py
index 3123c4745..2a4141bd4 100644
--- a/litellm/tests/test_secrets.py
+++ b/litellm/tests/test_secrets.py
@@ -19,7 +19,7 @@ messages = [{ "content": user_message,"role": "user"}]
 
 def test_completion_azure():
     try:
-        response = completion(model="chatgpt-test", messages=messages, azure=True)
+        response = completion(model="gpt-3.5-turbo", deployment_id="chatgpt-test", messages=messages, azure=True)
         # Add any assertions here to check the response
         print(response)
     except Exception as e:
diff --git a/pyproject.toml b/pyproject.toml
index 9bf71df72..a83be2b3b 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "litellm"
-version = "0.1.350"
+version = "0.1.351"
 description = "Library to easily interface with LLM API providers"
 authors = ["BerriAI"]
 license = "MIT License"