diff --git a/tests/unit_tests/callbacks/fake_callback_handler.py b/tests/unit_tests/callbacks/fake_callback_handler.py
index 03bd07c6c19..77da5a63c91 100644
--- a/tests/unit_tests/callbacks/fake_callback_handler.py
+++ b/tests/unit_tests/callbacks/fake_callback_handler.py
@@ -2,7 +2,7 @@
 from typing import Any, Dict, List
 
 from langchain.callbacks.base import BaseCallbackHandler
-from langchain.schema import AgentAction, LLMResult, AgentFinish
+from langchain.schema import AgentAction, AgentFinish, LLMResult
 
 
 class FakeCallbackHandler(BaseCallbackHandler):
@@ -66,4 +66,3 @@ class FakeCallbackHandler(BaseCallbackHandler):
     def on_agent_end(self, finish: AgentFinish, **kwargs: Any) -> None:
         """Run when agent ends running."""
         self.ends += 1
-