diff --git a/tests/traces/test_unified_trace_controller.py b/tests/traces/test_unified_trace_controller.py
index 80a35f11a..d6d46e83d 100644
--- a/tests/traces/test_unified_trace_controller.py
+++ b/tests/traces/test_unified_trace_controller.py
@@ -11,7 +11,6 @@ from crewai.traces.models import (
     FlowStepIO,
     LLMRequest,
     LLMResponse,
-    ToolCall,
 )
 from crewai.traces.unified_trace_controller import (
     UnifiedTraceController,