From 3a1c02ff3364fa68d6659282dc994e5c57fd1833 Mon Sep 17 00:00:00 2001
From: vbarda <vadym@langchain.dev>
Date: Thu, 15 May 2025 08:38:25 -0400
Subject: [PATCH] update for consistency

---
 libs/langgraph/langgraph/pregel/draw.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/libs/langgraph/langgraph/pregel/draw.py b/libs/langgraph/langgraph/pregel/draw.py
index 8e0df382b..bcae60d98 100644
--- a/libs/langgraph/langgraph/pregel/draw.py
+++ b/libs/langgraph/langgraph/pregel/draw.py
@@ -125,7 +125,7 @@ def draw_graph(
                                     edges.add((task.name, t[0], True, t[2]))
                             writes = [t for t in writes if t[0] != END]
                             conditionals.update(
-                                {(task.name, *t[:2]): t[2] for t in writes}
+                                {(task.name, t[0], t[1] or None): t[2] for t in writes}
                             )
                             task.config[CONF][CONFIG_KEY_SEND]([t[:2] for t in writes])
             # collect sources