From eeee10561a6adcffabf5b8bb13266ac4f62749ac Mon Sep 17 00:00:00 2001
From: VALADI K JAGANATHAN <info@orchidsasia.com>
Date: Wed, 13 Dec 2023 19:11:40 +0530
Subject: [PATCH] Update widgethider.js

---
 js/widgethider.js | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/js/widgethider.js b/js/widgethider.js
index 83b6d6d..82a9fa9 100644
--- a/js/widgethider.js
+++ b/js/widgethider.js
@@ -123,7 +123,7 @@ function handleInputModeWidgetsVisibility(node, inputModeValue) {
         }
     };
 
-    const inputModeVisibilityMap = nodeVisibilityMap[node.getTitle()];
+    const inputModeVisibilityMap = nodeVisibilityMap[node.comfyClass];
     
     if (!inputModeVisibilityMap || !inputModeVisibilityMap[inputModeValue]) return;
 
@@ -319,7 +319,7 @@ const nodeWidgetHandlers = {
 // In the main function where widgetLogic is called
 function widgetLogic(node, widget) {
     // Retrieve the handler for the current node title and widget name
-    const handler = nodeWidgetHandlers[node.getTitle()]?.[widget.name];
+    const handler = nodeWidgetHandlers[node.comfyClass]?.[widget.name];
     if (handler) {
         handler(node, widget);
     }