diff --git a/src_web/comfyui/node_mode_relay.ts b/src_web/comfyui/node_mode_relay.ts
index 922cfae..7bbb60e 100644
--- a/src_web/comfyui/node_mode_relay.ts
+++ b/src_web/comfyui/node_mode_relay.ts
@@ -96,6 +96,9 @@ class NodeModeRelay extends BaseCollectorNode {
   }
 
   override onDrawForeground(ctx: CanvasRenderingContext2D, canvas: LGraphCanvas): void {
+    if (this.flags?.collapsed) {
+      return;
+    }
     if (
         this.properties['on_muted_inputs'] !== 'MUTE' ||
         this.properties['on_bypassed_inputs'] !== 'BYPASS' ||
diff --git a/web/comfyui/node_mode_relay.js b/web/comfyui/node_mode_relay.js
index 4c3d6be..5ede7a7 100644
--- a/web/comfyui/node_mode_relay.js
+++ b/web/comfyui/node_mode_relay.js
@@ -44,6 +44,10 @@ class NodeModeRelay extends BaseCollectorNode {
         this.properties['on_any_active_inputs'] = 'ACTIVE';
     }
     onDrawForeground(ctx, canvas) {
+        var _a;
+        if ((_a = this.flags) === null || _a === void 0 ? void 0 : _a.collapsed) {
+            return;
+        }
         if (this.properties['on_muted_inputs'] !== 'MUTE' ||
             this.properties['on_bypassed_inputs'] !== 'BYPASS' ||
             this.properties['on_any_active_inputs'] != 'ACTIVE') {