diff --git a/vertexai/agent_engines/templates/langgraph.py b/vertexai/agent_engines/templates/langgraph.py
index 9f5295d81a..aff8b305b0 100644
--- a/vertexai/agent_engines/templates/langgraph.py
+++ b/vertexai/agent_engines/templates/langgraph.py
@@ -554,7 +554,7 @@ def query(
         Returns:
             The output of querying the Agent with the given input and config.
         """
-        from langchain.load import dump as langchain_load_dump
+        from langchain_core.load import dump as langchain_load_dump
 
         if isinstance(input, str):
             input = {"input": input}
@@ -587,7 +587,7 @@ def stream_query(
         Yields:
             The output of querying the Agent with the given input and config.
         """
-        from langchain.load import dump as langchain_load_dump
+        from langchain_core.load import dump as langchain_load_dump
 
         if isinstance(input, str):
             input = {"input": input}