diff --git a/tests/tools/test_search_tools.py b/tests/tools/test_search_tools.py
index fc7c1944a..a4b900d53 100644
--- a/tests/tools/test_search_tools.py
+++ b/tests/tools/test_search_tools.py
@@ -13,9 +13,8 @@ import pytest
 from nanobot.agent.loop import AgentLoop
 from nanobot.agent.subagent import SubagentManager, SubagentStatus
 from nanobot.agent.tools.search import FindFilesTool, GrepTool
-from nanobot.agent.tools.web import WebSearchTool
+from nanobot.agent.tools.web import WebSearchConfig, WebSearchTool
 from nanobot.bus.queue import MessageBus
-from nanobot.config.schema import WebSearchConfig
 
 
 @pytest.mark.asyncio
diff --git a/tests/tools/test_web_fetch_security.py b/tests/tools/test_web_fetch_security.py
index 6fb1d0f64..b2c245b8d 100644
--- a/tests/tools/test_web_fetch_security.py
+++ b/tests/tools/test_web_fetch_security.py
@@ -10,8 +10,7 @@ import httpx
 import pytest
 
 from nanobot.agent.tools import web as web_module
-from nanobot.agent.tools.web import WebFetchTool
-from nanobot.config.schema import WebFetchConfig
+from nanobot.agent.tools.web import WebFetchConfig, WebFetchTool
 from nanobot.security.workspace_access import (
     bind_workspace_scope,
     build_workspace_scope,
diff --git a/tests/tools/test_web_search_tool.py b/tests/tools/test_web_search_tool.py
index 1fd81f0ce..fcf015820 100644
--- a/tests/tools/test_web_search_tool.py
+++ b/tests/tools/test_web_search_tool.py
@@ -3,8 +3,7 @@
 import httpx
 import pytest
 
-from nanobot.agent.tools.web import WebSearchTool
-from nanobot.config.schema import WebSearchConfig
+from nanobot.agent.tools.web import WebSearchConfig, WebSearchTool
 
 
 def _tool(