From 97f3c4788533ca24357a0483e9a9848e66e1e023 Mon Sep 17 00:00:00 2001
From: VALADI K JAGANATHAN <info@orchidsasia.com>
Date: Wed, 13 Dec 2023 19:08:44 +0530
Subject: [PATCH] Update appearance.js

---
 js/appearance.js | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/js/appearance.js b/js/appearance.js
index 1e87498..2e41be5 100644
--- a/js/appearance.js
+++ b/js/appearance.js
@@ -77,9 +77,9 @@ const ext = {
     name: "efficiency.appearance",
 
     nodeCreated(node) {
-        const title = node.getTitle();
-        if (NODE_COLORS.hasOwnProperty(title)) {
-            let colorKey = NODE_COLORS[title];
+        const nclass = node.comfyClass;
+        if (NODE_COLORS.hasOwnProperty(nclass)) {
+            let colorKey = NODE_COLORS[nclass];
 
             if (colorKey === "random") {
                 // Check for a valid color key before popping
@@ -96,4 +96,4 @@ const ext = {
     }
 };
 
-app.registerExtension(ext);
\ No newline at end of file
+app.registerExtension(ext);