diff --git a/phi/agent/agent.py b/phi/agent/agent.py
index 9cba61c55..21b456fde 100644
--- a/phi/agent/agent.py
+++ b/phi/agent/agent.py
@@ -31,7 +31,7 @@
 from phi.reasoning.step import ReasoningStep, ReasoningSteps, NextAction
 from phi.run.response import RunEvent, RunResponse, RunResponseExtraData
 from phi.knowledge.agent import AgentKnowledge
-from phi.model import Model
+from phi.model.base import Model
 from phi.model.message import Message, MessageReferences
 from phi.model.response import ModelResponse, ModelResponseEvent
 from phi.memory.agent import AgentMemory, MemoryRetrieval, Memory, AgentRun, SessionSummary  # noqa: F401
diff --git a/phi/model/__init__.py b/phi/model/__init__.py
index 00c37db69..e69de29bb 100644
--- a/phi/model/__init__.py
+++ b/phi/model/__init__.py
@@ -1 +0,0 @@
-from phi.model.base import Model
diff --git a/pyproject.toml b/pyproject.toml
index fa7685340..b3c1fa0a8 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "phidata"
-version = "2.6.5"
+version = "2.6.6"
 description = "Build multi-modal Agents with memory, knowledge and tools."
 requires-python = ">=3.7"
 readme = "README.md"