diff --git a/tests/unit/llms/bedrock/search/test_agentcore_search_transformation.py b/tests/unit/llms/bedrock/search/test_agentcore_search_transformation.py
index 98d1da11c84..aaab56f00e5 100644
--- a/tests/unit/llms/bedrock/search/test_agentcore_search_transformation.py
+++ b/tests/unit/llms/bedrock/search/test_agentcore_search_transformation.py
@@ -7,11 +7,11 @@ the sharded CI (coverage collection runs against this tree).
 """
 
 import json
-from typing import Final
 import os
+from typing import Final
+from unittest.mock import AsyncMock, MagicMock, patch
 
 import pytest
-from unittest.mock import AsyncMock, patch, MagicMock
 
 import litellm
 from litellm.llms.bedrock.search.transformation import (