diff --git a/libs/langchain_v1/tests/unit_tests/agents/test_agent_streaming.py b/libs/langchain_v1/tests/unit_tests/agents/test_agent_streaming.py
index 2ae118854c2..af21bc95496 100644
--- a/libs/langchain_v1/tests/unit_tests/agents/test_agent_streaming.py
+++ b/libs/langchain_v1/tests/unit_tests/agents/test_agent_streaming.py
@@ -10,14 +10,13 @@ from langchain_core.tools import tool
 from langgraph.stream import EventLog, StreamTransformer
 
 from langchain.agents import create_agent
+from langchain.tools import ToolRuntime  # noqa: TC001
 from tests.unit_tests.agents.model import FakeToolCallingModel
 
 if TYPE_CHECKING:
     from langgraph.prebuilt._tool_call_stream import ToolCallStream
     from langgraph.stream._types import ProtocolEvent
 
-    from langchain.tools import ToolRuntime
-
 
 @tool
 def echo(text: str) -> str: