diff --git a/src/extensions/core/nodeBadge.ts b/src/extensions/core/nodeBadge.ts
index a1b2a5b71..9a3ae5e1e 100644
--- a/src/extensions/core/nodeBadge.ts
+++ b/src/extensions/core/nodeBadge.ts
@@ -15,6 +15,10 @@ import { useNodeDefStore } from '@/stores/nodeDefStore'
 function getNodeSource(node: ComfyLGraphNode): NodeSource | null {
   const nodeDef = (node.constructor as typeof ComfyLGraphNode)
     .nodeData as ComfyNodeDef
+  // Frontend-only nodes don't have nodeDef
+  if (!nodeDef) {
+    return null
+  }
   const nodeDefStore = useNodeDefStore()
   return nodeDefStore.nodeDefsByName[nodeDef.name]?.nodeSource ?? null
 }