diff --git a/tests/tools/test_tool_usage.py b/tests/tools/test_tool_usage.py
index a0f364fac..76ec5eefb 100644
--- a/tests/tools/test_tool_usage.py
+++ b/tests/tools/test_tool_usage.py
@@ -8,6 +8,7 @@ from pydantic import BaseModel, Field
 from crewai import Agent, Task
 from crewai.tools import BaseTool
 from crewai.tools.tool_usage import ToolUsage
+from crewai.tools.tool_calling import ToolCalling
 
 
 class RandomNumberToolInput(BaseModel):