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: