diff --git a/__init__.py b/__init__.py
index 48d8e29b..20cd4f69 100644
--- a/__init__.py
+++ b/__init__.py
@@ -29,7 +29,7 @@ except:
     print(f"[WARN] ComfyUI-Manager: Your ComfyUI version is outdated. Please update to the latest version.")
 
 
-version = [2, 3, 1]
+version = [2, 3, 2]
 version_str = f"V{version[0]}.{version[1]}" + (f'.{version[2]}' if len(version) > 2 else '')
 print(f"### Loading: ComfyUI-Manager ({version_str})")
 
diff --git a/js/comfyui-manager.js b/js/comfyui-manager.js
index 2452dbea..341f2a09 100644
--- a/js/comfyui-manager.js
+++ b/js/comfyui-manager.js
@@ -1271,7 +1271,10 @@ app.registerExtension({
 	async nodeCreated(node, app) {
 		if(!node.badge_enabled) {
 			node.getNickname = function () { return getNickname(node, node.comfyClass.trim()) };
-			const orig = node.__proto__.onDrawForeground;
+			let orig = node.onDrawForeground;
+			if(!orig)
+				orig = node.__proto__.onDrawForeground;
+
 			node.onDrawForeground = function (ctx) {
 				drawBadge(node, orig, arguments)
 			};