diff --git a/libs/langgraph/langgraph/pregel/debug.py b/libs/langgraph/langgraph/pregel/debug.py
index 5a317bb26..389c59e81 100644
--- a/libs/langgraph/langgraph/pregel/debug.py
+++ b/libs/langgraph/langgraph/pregel/debug.py
@@ -110,7 +110,7 @@ def map_debug_task_results(
 
 
 def map_debug_checkpoint(
-    step: int,  # TODO remove arg, it's in metadata
+    step: int,
     config: RunnableConfig,
     channels: Mapping[str, BaseChannel],
     stream_channels: Union[str, Sequence[str]],