diff --git a/swarms/models/__init__.py b/swarms/models/__init__.py
index 2a955ecc..973c8216 100644
--- a/swarms/models/__init__.py
+++ b/swarms/models/__init__.py
@@ -1,4 +1,5 @@
 from swarms.models.anthropic import Anthropic
 from swarms.models.petals import Petals
 from swarms.models.mistral import Mistral
-from swarms.models.openai_llm import OpenAIModel
+# from swarms.models.openai_llm import OpenAIModel
+from swarms.models.openai_models import OpenAI, AzureOpenAI, OpenAIChat
\ No newline at end of file