diff --git a/vision_agent/__init__.py b/vision_agent/__init__.py
index 3704c363..360b8a78 100644
--- a/vision_agent/__init__.py
+++ b/vision_agent/__init__.py
@@ -1,5 +1,5 @@
+from .agent import Agent
 from .data import DataStore, build_data_store
 from .emb import Embedder, OpenAIEmb, SentenceTransformerEmb, get_embedder
 from .llm import LLM, OpenAILLM
 from .lmm import LMM, LLaVALMM, OpenAILMM, get_lmm
-from .agent import Agent
diff --git a/vision_agent/agent/__init__.py b/vision_agent/agent/__init__.py
index aec05098..f0954251 100644
--- a/vision_agent/agent/__init__.py
+++ b/vision_agent/agent/__init__.py
@@ -1,3 +1,3 @@
 from .agent import Agent
-from .reflexion import Reflexion
 from .easytool import EasyTool
+from .reflexion import Reflexion
diff --git a/vision_agent/tools/__init__.py b/vision_agent/tools/__init__.py
index 67921aae..7e5636c2 100644
--- a/vision_agent/tools/__init__.py
+++ b/vision_agent/tools/__init__.py
@@ -1,10 +1,2 @@
 from .prompts import CHOOSE_PARAMS, SYSTEM_PROMPT
-from .tools import (
-    CLIP,
-    TOOLS,
-    Counter,
-    Crop,
-    GroundingDINO,
-    GroundingSAM,
-    Tool,
-)
+from .tools import CLIP, TOOLS, Counter, Crop, GroundingDINO, GroundingSAM, Tool