diff --git a/litellm/llms/AzureOpenAI/azure.py b/litellm/llms/AzureOpenAI/azure.py
index a14644e1875..098086c8b98 100644
--- a/litellm/llms/AzureOpenAI/azure.py
+++ b/litellm/llms/AzureOpenAI/azure.py
@@ -2291,7 +2291,7 @@ class AzureAssistantsAPI(BaseLLM):
         """
         Here's an example:
         ```
-        from litellm.llms.openai import OpenAIAssistantsAPI, MessageData
+        from litellm.llms.OpenAI.openai import OpenAIAssistantsAPI, MessageData
 
         # create thread
         message: MessageData = {"role": "user", "content": "Hey, how's it going?"}
diff --git a/litellm/llms/OpenAI/openai.py b/litellm/llms/OpenAI/openai.py
index 8d112f52ae9..8021ccd59e4 100644
--- a/litellm/llms/OpenAI/openai.py
+++ b/litellm/llms/OpenAI/openai.py
@@ -3056,7 +3056,7 @@ class OpenAIAssistantsAPI(BaseLLM):
         """
         Here's an example:
         ```
-        from litellm.llms.openai import OpenAIAssistantsAPI, MessageData
+        from litellm.llms.OpenAI.openai import OpenAIAssistantsAPI, MessageData
 
         # create thread
         message: MessageData = {"role": "user", "content": "Hey, how's it going?"}
diff --git a/litellm/llms/vertex_ai_and_google_ai_studio/vertex_ai_partner_models/main.py b/litellm/llms/vertex_ai_and_google_ai_studio/vertex_ai_partner_models/main.py
index 60c1fa607d8..69909765e82 100644
--- a/litellm/llms/vertex_ai_and_google_ai_studio/vertex_ai_partner_models/main.py
+++ b/litellm/llms/vertex_ai_and_google_ai_studio/vertex_ai_partner_models/main.py
@@ -81,7 +81,7 @@ class VertexAIPartnerModels(BaseLLM):
             from google.cloud import aiplatform
 
             from litellm.llms.databricks import DatabricksChatCompletion
-            from litellm.llms.openai import OpenAIChatCompletion
+            from litellm.llms.OpenAI.openai import OpenAIChatCompletion
             from litellm.llms.text_completion_codestral import CodestralTextCompletion
             from litellm.llms.vertex_ai_and_google_ai_studio.gemini.vertex_and_google_ai_studio_gemini import (
                 VertexLLM,