From 36e15f2507ee81e27140cf15ffcda40070968928 Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Sun, 31 Dec 2023 05:05:14 -0500
Subject: [PATCH] Reregister nodes when pressing refresh button.

---
 web/scripts/app.js | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

diff --git a/web/scripts/app.js b/web/scripts/app.js
index 62b71c0a..7353f5a3 100644
--- a/web/scripts/app.js
+++ b/web/scripts/app.js
@@ -2020,12 +2020,8 @@ export class ComfyApp {
 	async refreshComboInNodes() {
 		const defs = await api.getNodeDefs();
 
-		for(const nodeId in LiteGraph.registered_node_types) {
-			const node = LiteGraph.registered_node_types[nodeId];
-			const nodeDef = defs[nodeId];
-			if(!nodeDef) continue;
-
-			node.nodeData = nodeDef;
+		for (const nodeId in defs) {
+			this.registerNodeDef(nodeId, defs[nodeId]);
 		}
 
 		for(let nodeNum in this.graph._nodes) {