diff --git a/tests/integration/test_tool_group_integration.py b/tests/integration/test_tool_group_integration.py
index 9b8673f7..cac37590 100644
--- a/tests/integration/test_tool_group_integration.py
+++ b/tests/integration/test_tool_group_integration.py
@@ -7,8 +7,14 @@ tool filtering and execution in the ReAct agent service.
 
 import pytest
 import json
+import sys
+import os
 from unittest.mock import Mock, AsyncMock, patch
 
+# Add trustgraph paths for imports
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', '..', 'trustgraph-base'))
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', '..', 'trustgraph-flow'))
+
 from trustgraph.schema import AgentRequest, AgentResponse, AgentStep
 from trustgraph.agent.react.service import Processor
 from trustgraph.agent.react.types import Tool, Argument