From 273eae1e8d8dadba8f6c1c604004dfdf973dc276 Mon Sep 17 00:00:00 2001
From: filtered <176114999+webfiltered@users.noreply.github.com>
Date: Wed, 28 May 2025 10:12:13 +1000
Subject: [PATCH] Fix custom widget drawWidget never called (#1068)

---
 src/widgets/widgetMap.ts | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/widgets/widgetMap.ts b/src/widgets/widgetMap.ts
index 4f5b5b3f2..07dc9033a 100644
--- a/src/widgets/widgetMap.ts
+++ b/src/widgets/widgetMap.ts
@@ -59,6 +59,8 @@ export function toConcreteWidget<TWidget extends IWidget | IBaseWidget>(
   node: LGraphNode,
   wrapLegacyWidgets = true,
 ): WidgetTypeMap[TWidget["type"]] | undefined {
+  if (widget instanceof BaseWidget) return widget
+
   // Assertion: TypeScript has no concept of "all strings except X"
   type RemoveBaseWidgetType<T> = T extends { type: TWidgetType } ? T : never
   const narrowedWidget = widget as RemoveBaseWidgetType<TWidget>