From 22ce64ad0962a8c23277c0d731769da8a5f582fa Mon Sep 17 00:00:00 2001
From: Christian Byrne <cbyrne@comfy.org>
Date: Thu, 29 Jan 2026 12:01:31 -0800
Subject: [PATCH] Update src/stores/modelToNodeStore.ts

Co-authored-by: Alexander Brown <drjkl@comfy.org>
---
 src/stores/modelToNodeStore.ts | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/src/stores/modelToNodeStore.ts b/src/stores/modelToNodeStore.ts
index 2355da230..7950fd94e 100644
--- a/src/stores/modelToNodeStore.ts
+++ b/src/stores/modelToNodeStore.ts
@@ -85,10 +85,12 @@ export const useModelToNodeStore = defineStore('modelToNode', () => {
     if (exactMatch && exactMatch.length > 0) return exactMatch
 
     const topLevel = modelType.split('/')[0]
-    if (topLevel !== modelType) {
-      const fallback = modelToNodeMap.value[topLevel]
-      if (fallback && fallback.length > 0) return fallback
-    }
+    if (topLevel === modelType) return undefined
+
+    const fallback = modelToNodeMap.value[topLevel]
+
+    if (fallback && fallback.length > 0) return fallback
+    
     return undefined
   }