From 16878244530c07c6fdad08c4b898b818c44c0d6f Mon Sep 17 00:00:00 2001
From: Lucas Oliveira <lucas.edu.oli@hotmail.com>
Date: Mon, 31 Jul 2023 14:01:54 -0300
Subject: [PATCH] Optimized code

---
 src/frontend/src/utils/utils.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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)),
       });
     }