diff --git a/src/services/extensionService.ts b/src/services/extensionService.ts
index a8c1a1901..6bca3c031 100644
--- a/src/services/extensionService.ts
+++ b/src/services/extensionService.ts
@@ -82,11 +82,10 @@ export const useExtensionService = () => {
     }
 
     if (extension.getCustomVueWidgets) {
+      const getVueWidgets = extension.getCustomVueWidgets
       void (async () => {
-        if (extension.getCustomVueWidgets) {
-          const vueWidgets = await extension.getCustomVueWidgets(app)
-          registerVueWidgets(vueWidgets)
-        }
+        const vueWidgets = await getVueWidgets(app)
+        registerVueWidgets(vueWidgets)
       })()
     }
 
diff --git a/src/types/comfy.ts b/src/types/comfy.ts
index d1c5f2139..b89956780 100644
--- a/src/types/comfy.ts
+++ b/src/types/comfy.ts
@@ -31,7 +31,7 @@ export interface VueWidgetDefinition {
   aliases?: string[]
 }
 
-type VueWidgets = Record<string, VueWidgetDefinition>
+export type VueWidgets = Record<string, VueWidgetDefinition>
 
 export interface AboutPageBadge {
   label: string