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);