diff --git a/frontend/layout/lib/layoutNode.ts b/frontend/layout/lib/layoutNode.ts
index 319edcf0..95bc3b7a 100644
--- a/frontend/layout/lib/layoutNode.ts
+++ b/frontend/layout/lib/layoutNode.ts
@@ -186,7 +186,6 @@ export function balanceNode(
     walkNodes(
         node,
         (node) => {
-            beforeWalkCallback?.(node);
             if (!validateNode(node)) throw new Error("Invalid node");
             node.children = node.children?.flatMap((child) => {
                 if (child.flexDirection === node.flexDirection) {
@@ -198,6 +197,7 @@ export function balanceNode(
                 if (child.children?.length === 0) return;
                 return child;
             });
+            beforeWalkCallback?.(node);
         },
         (node) => {
             node.children = node.children?.filter((v) => v);