From abf93d2c119461efaf1d680e3aae11e63938fa89 Mon Sep 17 00:00:00 2001
From: Christian Byrne <cbyrne@comfy.org>
Date: Tue, 29 Jul 2025 18:20:15 -0700
Subject: [PATCH] [fix] Update canvas when removing subgraph IO slots to
 re-render links (#1173)

---
 src/subgraph/SubgraphNode.ts | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/subgraph/SubgraphNode.ts b/src/subgraph/SubgraphNode.ts
index b3597bbb9..ba985b7ee 100644
--- a/src/subgraph/SubgraphNode.ts
+++ b/src/subgraph/SubgraphNode.ts
@@ -68,6 +68,7 @@ export class SubgraphNode extends LGraphNode implements BaseLGraph {
       if (widget) this.ensureWidgetRemoved(widget)
 
       this.removeInput(e.detail.index)
+      this.setDirtyCanvas(true, true)
     }, { signal })
 
     subgraphEvents.addEventListener("output-added", (e) => {
@@ -77,6 +78,7 @@ export class SubgraphNode extends LGraphNode implements BaseLGraph {
 
     subgraphEvents.addEventListener("removing-output", (e) => {
       this.removeOutput(e.detail.index)
+      this.setDirtyCanvas(true, true)
     }, { signal })
 
     subgraphEvents.addEventListener("renaming-input", (e) => {