diff --git a/libs/langchain_v1/langchain/agents/middleware_agent.py b/libs/langchain_v1/langchain/agents/middleware_agent.py
index 718e4a692e6..30e4bbc331a 100644
--- a/libs/langchain_v1/langchain/agents/middleware_agent.py
+++ b/libs/langchain_v1/langchain/agents/middleware_agent.py
@@ -33,7 +33,7 @@ from langchain.agents.structured_output import (
     StructuredOutputValidationError,
     ToolStrategy,
 )
-from langchain.agents.tool_node import ToolNode
+from langchain.tools import ToolNode
 from langchain.chat_models import init_chat_model
 
 STRUCTURED_OUTPUT_ERROR_TEMPLATE = "Error: {error}\n Please fix your mistakes."
diff --git a/libs/langchain_v1/langchain/agents/react_agent.py b/libs/langchain_v1/langchain/agents/react_agent.py
index d9392c47527..21c12ce6cb9 100644
--- a/libs/langchain_v1/langchain/agents/react_agent.py
+++ b/libs/langchain_v1/langchain/agents/react_agent.py
@@ -54,8 +54,8 @@ from langchain.agents.structured_output import (
     StructuredOutputValidationError,
     ToolStrategy,
 )
-from langchain.agents.tool_node import ToolNode
 from langchain.chat_models import init_chat_model
+from langchain.tools import ToolNode
 
 if TYPE_CHECKING:
     from langchain_core.tools import BaseTool
@@ -66,7 +66,7 @@ if TYPE_CHECKING:
     from langchain.agents._internal._typing import (
         SyncOrAsync,
     )
-    from langchain.agents.types import AgentMiddleware
+    from langchain.agents.middleware import AgentMiddleware
 
 StructuredResponseT = TypeVar("StructuredResponseT", default=None)
 
diff --git a/libs/langchain_v1/tests/unit_tests/tools/test_imports.py b/libs/langchain_v1/tests/unit_tests/tools/test_imports.py
index 5ed7ffe950f..1b885d60b5c 100644
--- a/libs/langchain_v1/tests/unit_tests/tools/test_imports.py
+++ b/libs/langchain_v1/tests/unit_tests/tools/test_imports.py
@@ -6,6 +6,7 @@ EXPECTED_ALL = {
     "InjectedToolCallId",
     "ToolException",
     "tool",
+    "ToolNode",
 }