diff --git a/autogpt_platform/backend/backend/copilot/tools/find_block_test.py b/autogpt_platform/backend/backend/copilot/tools/find_block_test.py
index f172605374..ef4c63e7f9 100644
--- a/autogpt_platform/backend/backend/copilot/tools/find_block_test.py
+++ b/autogpt_platform/backend/backend/copilot/tools/find_block_test.py
@@ -76,12 +76,12 @@ class TestFindBlockFiltering:
             }.get(block_id)
 
         with patch(
-            "backend.api.features.chat.tools.find_block.unified_hybrid_search",
+            "backend.copilot.tools.find_block.unified_hybrid_search",
             new_callable=AsyncMock,
             return_value=(search_results, 2),
         ):
             with patch(
-                "backend.api.features.chat.tools.find_block.get_block",
+                "backend.copilot.tools.find_block.get_block",
                 side_effect=mock_get_block,
             ):
                 tool = FindBlockTool()
@@ -120,12 +120,12 @@ class TestFindBlockFiltering:
             }.get(block_id)
 
         with patch(
-            "backend.api.features.chat.tools.find_block.unified_hybrid_search",
+            "backend.copilot.tools.find_block.unified_hybrid_search",
             new_callable=AsyncMock,
             return_value=(search_results, 2),
         ):
             with patch(
-                "backend.api.features.chat.tools.find_block.get_block",
+                "backend.copilot.tools.find_block.get_block",
                 side_effect=mock_get_block,
             ):
                 tool = FindBlockTool()
diff --git a/autogpt_platform/backend/backend/copilot/tools/run_block_test.py b/autogpt_platform/backend/backend/copilot/tools/run_block_test.py
index 1a860e24d6..1f9e2a397e 100644
--- a/autogpt_platform/backend/backend/copilot/tools/run_block_test.py
+++ b/autogpt_platform/backend/backend/copilot/tools/run_block_test.py
@@ -39,7 +39,7 @@ class TestRunBlockFiltering:
         input_block = make_mock_block("input-block-id", "Input Block", BlockType.INPUT)
 
         with patch(
-            "backend.api.features.chat.tools.run_block.get_block",
+            "backend.copilot.tools.run_block.get_block",
             return_value=input_block,
         ):
             tool = RunBlockTool()
@@ -65,7 +65,7 @@ class TestRunBlockFiltering:
         )
 
         with patch(
-            "backend.api.features.chat.tools.run_block.get_block",
+            "backend.copilot.tools.run_block.get_block",
             return_value=smart_block,
         ):
             tool = RunBlockTool()
@@ -89,7 +89,7 @@ class TestRunBlockFiltering:
         )
 
         with patch(
-            "backend.api.features.chat.tools.run_block.get_block",
+            "backend.copilot.tools.run_block.get_block",
             return_value=standard_block,
         ):
             tool = RunBlockTool()
diff --git a/autogpt_platform/backend/test/agent_generator/test_library_agents.py b/autogpt_platform/backend/test/agent_generator/test_library_agents.py
index 2007708325..b0cd9f0c2c 100644
--- a/autogpt_platform/backend/test/agent_generator/test_library_agents.py
+++ b/autogpt_platform/backend/test/agent_generator/test_library_agents.py
@@ -151,7 +151,7 @@ class TestSearchMarketplaceAgentsForGeneration:
                 return_value=mock_response,
             ) as mock_search,
             patch(
-                "backend.api.features.chat.tools.agent_generator.core.get_store_listed_graphs",
+                "backend.copilot.tools.agent_generator.core.get_store_listed_graphs",
                 new_callable=AsyncMock,
                 return_value={"graph-123": mock_graph},
             ),