diff --git a/libs/langgraph/tests/test_messages_state.py b/libs/langgraph/tests/test_messages_state.py
index 0a1e78ecb..2994b2a47 100644
--- a/libs/langgraph/tests/test_messages_state.py
+++ b/libs/langgraph/tests/test_messages_state.py
@@ -20,7 +20,7 @@ from langgraph.graph.message import REMOVE_ALL_MESSAGES, MessagesState, push_mes
 from langgraph.graph.state import StateGraph
 from tests.messages import _AnyIdHumanMessage
 
-_, CORE_MINOR, CORE_PATCH = (int(v) for v in langchain_core.__version__.split("."))
+_, CORE_MINOR, CORE_PATCH = (int(v) for v in langchain_core.__version__.split(".")[:3])
 
 
 def test_add_single_message():
diff --git a/libs/prebuilt/langgraph/prebuilt/tool_node.py b/libs/prebuilt/langgraph/prebuilt/tool_node.py
index 3bf84243a..e7a88f309 100644
--- a/libs/prebuilt/langgraph/prebuilt/tool_node.py
+++ b/libs/prebuilt/langgraph/prebuilt/tool_node.py
@@ -522,7 +522,6 @@ class ToolNode(RunnableCallable):
             )
             return response
         elif isinstance(response, ToolMessageV1):
-            # TODO: Handle ToolMessageV1
             return response
         else:
             raise TypeError(
@@ -591,7 +590,6 @@ class ToolNode(RunnableCallable):
             )
             return response
         elif isinstance(response, ToolMessageV1):
-            # TODO: Handle ToolMessageV1
             return response
         else:
             raise TypeError(