diff --git a/libs/langgraph/langgraph/pregel/algo.py b/libs/langgraph/langgraph/pregel/algo.py
index 08dcf77e6..d1247dbc4 100644
--- a/libs/langgraph/langgraph/pregel/algo.py
+++ b/libs/langgraph/langgraph/pregel/algo.py
@@ -507,6 +507,7 @@ def prepare_single_task(
                         CONFIG_KEY_SCRATCHPAD: _scratchpad(
                             pending_writes,
                             task_id,
+                        ),
                         CONFIG_KEY_END: checkpoint["channel_values"].get(
                             "__end__", None
                         ),
@@ -745,7 +746,7 @@ def prepare_single_task(
                                 ),
                                 CONFIG_KEY_END: checkpoint["channel_values"].get(
                                     "__end__", None
-
+                                ),
                             },
                         ),
                         triggers,