From 001040abc7891603939fb9b221d16202635925d3 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Tue, 21 Oct 2025 20:27:37 +0300
Subject: [PATCH] Update nodes.py

---
 nodes.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/nodes.py b/nodes.py
index c9a25f8..78c149a 100644
--- a/nodes.py
+++ b/nodes.py
@@ -1839,7 +1839,7 @@ class WanVideoScheduler: #WIP
                 for x, y in zip(x_values, sigmas_np):
                     # Show all annotations if few steps, or just show split step annotations
                     show_annotation = len(sigmas_np) <= 10
-                    is_split_step = (start_idx > 0 and x == start_idx) or (end_idx != -1 and x == end_idx)
+                    is_split_step = (start_idx > 0 and x == start_idx) or (end_idx != -1 and x == end_idx + 1)
                     
                     if show_annotation or is_split_step:
                         color = 'orange'