diff --git a/swarms/models/__init__.py b/swarms/models/__init__.py
index eeccc4a1..5b759fb7 100644
--- a/swarms/models/__init__.py
+++ b/swarms/models/__init__.py
@@ -1,5 +1,5 @@
 # LLMs
-from swarms.models.base import AbstractLLM  # noqa: E402
+from swarms.models.base_llm import AbstractLLM  # noqa: E402
 from swarms.models.anthropic import Anthropic  # noqa: E402
 from swarms.models.petals import Petals  # noqa: E402
 from swarms.models.mistral import Mistral  # noqa: E402