From d60b3067662773b1c60c9f8eaca1d8c05b550609 Mon Sep 17 00:00:00 2001
From: "Dr.Lt.Data" <dr.lt.data@gmail.com>
Date: Sat, 17 Jun 2023 17:07:36 +0900
Subject: [PATCH] bugfix: LoopControl return type

---
 comfy_extras/nodes_execution_control.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/comfy_extras/nodes_execution_control.py b/comfy_extras/nodes_execution_control.py
index 3609b8257..50eb5a33b 100644
--- a/comfy_extras/nodes_execution_control.py
+++ b/comfy_extras/nodes_execution_control.py
@@ -17,7 +17,11 @@ class LoopControl:
         else:
             current = kwargs['loopback_input']
 
-        return (kwargs['loop_condition'].get_next(kwargs['initial_input'], current), )
+        result = kwargs['loop_condition'].get_next(kwargs['initial_input'], current)
+        if result is None:
+            return None
+        else:
+            return (result, )
 
 
 class CounterCondition: