diff --git a/libs/langgraph/langgraph/pregel/__init__.py b/libs/langgraph/langgraph/pregel/__init__.py
index e84c0b208..5b37e80fd 100644
--- a/libs/langgraph/langgraph/pregel/__init__.py
+++ b/libs/langgraph/langgraph/pregel/__init__.py
@@ -1485,7 +1485,7 @@ class Pregel(PregelProtocol):
                 ),
             )
 
-            for _, task in tasks:
+            for task_id, task in tasks:
                 if push_writes := [w for w in task.writes if w[0] == PUSH]:
                     checkpointer.put_writes(next_config, push_writes, task_id)
 
@@ -1818,7 +1818,7 @@ class Pregel(PregelProtocol):
                 ),
             )
 
-            for _, task in tasks:
+            for task_id, task in tasks:
                 # save push writes
                 if push_writes := [w for w in task.writes if w[0] == PUSH]:
                     await checkpointer.aput_writes(next_config, push_writes, task_id)