diff --git a/tests/flow/test_flow_tool_result.py b/tests/flow/test_flow_tool_result.py
index 1d00241a8..51a1be494 100644
--- a/tests/flow/test_flow_tool_result.py
+++ b/tests/flow/test_flow_tool_result.py
@@ -1,11 +1,13 @@
 import pytest
-from unittest.mock import patch, MagicMock
-from crewai import Agent, Task, Crew, Flow
-from crewai.flow import start, listen
-from crewai.tools import BaseTool
 from typing import Type
+from unittest.mock import MagicMock, patch
+
 from pydantic import BaseModel, Field
 
+from crewai import Agent, Crew, Flow, Task
+from crewai.flow import listen, start
+from crewai.tools import BaseTool
+
 class TestToolInput(BaseModel):
     query: str = Field(..., description='Query to process')