diff --git a/flaml/autogen/oai/completion.py b/flaml/autogen/oai/completion.py
index f98ccd8a8..e7b4a5031 100644
--- a/flaml/autogen/oai/completion.py
+++ b/flaml/autogen/oai/completion.py
@@ -236,7 +236,7 @@ class Completion(openai_Completion):
                 request_timeout = min(request_timeout, time_left)
                 sleep(cls.retry_time)
             except InvalidRequestError:
-                if "azure" == config.get("api_type", openai.api_type) and "model" in config:
+                if "azure" in config.get("api_type", openai.api_type) and "model" in config:
                     # azure api uses "engine" instead of "model"
                     config["engine"] = config.pop("model").replace("gpt-3.5-turbo", "gpt-35-turbo")
                 else: