From 8d622e65196aaa99bbb6fb211d5829d6cf5e356a Mon Sep 17 00:00:00 2001
From: Austin Mroz <austin@comfy.org>
Date: Thu, 28 Aug 2025 12:02:50 -0700
Subject: [PATCH] Improve code quality

---
 src/lib/litegraph/src/LGraphNode.ts | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/src/lib/litegraph/src/LGraphNode.ts b/src/lib/litegraph/src/LGraphNode.ts
index 56f593179..834a064b6 100644
--- a/src/lib/litegraph/src/LGraphNode.ts
+++ b/src/lib/litegraph/src/LGraphNode.ts
@@ -3998,12 +3998,10 @@ export class LGraphNode
    */
   #arrangeWidgets(widgetStartY: number): void {
     if (!this.widgets || !this.widgets.length) return
-    for (let i = 0; i < this.widgets.length; i++) {
-      if (!this.widgets[i]) {
-        console.error('Removing undefined widget in', this)
-        this.widgets.splice(i, 1)
-        i--
-      }
+    const realWidgets = this.widgets.filter((w) => w)
+    if (realWidgets.length != this.widgets.length) {
+      console.error('Removing undefined widgets from', this)
+      this.widgets.splice(0, this.widgets.length, ...realWidgets)
     }
 
     const bodyHeight = this.bodyHeight