diff --git a/apps/ai-game-creator-shell/src/view/ui-editor/index.tsx b/apps/ai-game-creator-shell/src/view/ui-editor/index.tsx
index f6786c58b..8d82cf444 100644
--- a/apps/ai-game-creator-shell/src/view/ui-editor/index.tsx
+++ b/apps/ai-game-creator-shell/src/view/ui-editor/index.tsx
@@ -80,8 +80,9 @@ export default function UiEditorPage({
   const [showFrame, setShowFrame] = useState(true);
   const [showOriginImage, setShowOriginImage] = useState(true);
   const [showComponent, setShowComponent] = useState(false);
-  const overview =
-    session.input.activeStep === 'structure-recognition' ? (
+  let overview: ReactNode;
+  if (session.input.activeStep === 'structure-recognition') {
+    overview = (
       <RecognitionOverview
         uiTrees={session.input.uiTrees}
         onFocusStatusNode={(treeId, nodeId) => {
@@ -89,7 +90,9 @@ export default function UiEditorPage({
           session.input.highlightStatusField(nodeId, 'layout_status');
         }}
       />
-    ) : session.input.activeStep === 'asset-separation' ? (
+    );
+  } else if (session.input.activeStep === 'asset-separation') {
+    overview = (
       <SeparationOverview
         uiTrees={session.input.uiTrees}
         sprites={session.input.sprites}
@@ -99,13 +102,16 @@ export default function UiEditorPage({
           session.input.highlightStatusField(nodeId, 'component_status');
         }}
       />
-    ) : (
+    );
+  } else {
+    overview = (
       <ImportOverview
         images={session.input.images}
         sprites={session.input.sprites}
         uiTrees={session.input.uiTrees}
       />
     );
+  }
   const saveDisabled =
     session.save.isSaving ||
     session.save.isGenerating ||