diff --git a/web/src/views/classification/ModelSelectionView.tsx b/web/src/views/classification/ModelSelectionView.tsx
index 095d5b63a0..5dd5418f12 100644
--- a/web/src/views/classification/ModelSelectionView.tsx
+++ b/web/src/views/classification/ModelSelectionView.tsx
@@ -271,17 +271,16 @@ function ModelCard({ config, onClick, onUpdate, onDelete }: ModelCardProps) {
       (key) => key != "none" && key.toLowerCase() != "unknown",
     );
 
-    if (keys.length === 0) {
+    const selectedKey = keys.find(
+      (key) => (dataset.categories[key]?.length ?? 0) > 0,
+    );
+
+    if (!selectedKey) {
       return undefined;
     }
 
-    const selectedKey = keys[0];
     const images = dataset.categories[selectedKey];
 
-    if (!images || images.length === 0) {
-      return undefined;
-    }
-
     return {
       name: selectedKey,
       img: images[0],