diff --git a/libs/langgraph/langgraph/managed/is_last_step.py b/libs/langgraph/langgraph/managed/is_last_step.py
index 6ffa4df16..427361351 100644
--- a/libs/langgraph/langgraph/managed/is_last_step.py
+++ b/libs/langgraph/langgraph/managed/is_last_step.py
@@ -1,7 +1,7 @@
 from typing import Annotated
 
 from langgraph.managed.base import ManagedValue
-from langgraph.pregel._scratchpad import PregelScratchpad
+from langgraph.types import PregelScratchpad
 
 __all__ = ("IsLastStep", "RemainingStepsManager")
 
@@ -22,3 +22,4 @@ class RemainingStepsManager(ManagedValue[int]):
 
 
 RemainingSteps = Annotated[int, RemainingStepsManager]
+