diff --git a/litellm/__pycache__/main.cpython-311.pyc b/litellm/__pycache__/main.cpython-311.pyc
index 9d440cdce1e..70dd03ec3b1 100644
Binary files a/litellm/__pycache__/main.cpython-311.pyc and b/litellm/__pycache__/main.cpython-311.pyc differ
diff --git a/litellm/main.py b/litellm/main.py
index 8f4b15bca39..f456f030f8c 100644
--- a/litellm/main.py
+++ b/litellm/main.py
@@ -239,7 +239,7 @@ def completion(
             ]  # update the model to the actual value if an alias has been passed in
         model_response = ModelResponse()
 
-        if kwargs['azure'] == True: # don't remove flag check, to remain backwards compatible for repos like Codium
+        if kwargs.get('azure', False) == True: # don't remove flag check, to remain backwards compatible for repos like Codium
             custom_llm_provider="azure"
         if deployment_id != None: # azure llms 
                 model=deployment_id