diff --git a/src/frontend/src/utils/utils.ts b/src/frontend/src/utils/utils.ts
index aa6c69f13..462e72084 100644
--- a/src/frontend/src/utils/utils.ts
+++ b/src/frontend/src/utils/utils.ts
@@ -115,10 +115,10 @@ export function groupByFamily(data, baseClasses, left, flow?: NodeType[]) {
       const foundNode = checkedNodes.get(nodeData.type);
       checkedNodes.set(nodeData.type, {
         hasBaseClassInTemplate:
-          checkedNodes.get(nodeData.type)?.hasBaseClassInTemplate ||
+          foundNode?.hasBaseClassInTemplate ||
           Object.values(nodeData.node.template).some(checkBaseClass),
         hasBaseClassInBaseClasses:
-          checkedNodes.get(nodeData.type)?.hasBaseClassInBaseClasses ||
+          foundNode?.hasBaseClassInBaseClasses ||
           nodeData.node.base_classes.some((t) => baseClassesSet.has(t)),
       });
     }