diff --git a/web/src/components/overlay/detail/SearchDetailDialog.tsx b/web/src/components/overlay/detail/SearchDetailDialog.tsx
index d6571a7f3..9f430fef5 100644
--- a/web/src/components/overlay/detail/SearchDetailDialog.tsx
+++ b/web/src/components/overlay/detail/SearchDetailDialog.tsx
@@ -1595,7 +1595,8 @@ function ObjectDetailsTab({
                 const model = config?.classification?.custom?.[modelName];
                 if (!model) return null;
 
-                const classes = Object.keys(modelAttributes?.[modelName] ?? {});
+                const displayName = model.name || modelName;
+                const classes = modelAttributes?.[displayName] ?? [];
                 if (classes.length === 0) return null;
 
                 return (