diff --git a/apps/ai-game-creator-shell/src/view/ui-editor/components/preview/PreviewWorkspace.tsx b/apps/ai-game-creator-shell/src/view/ui-editor/components/preview/PreviewWorkspace.tsx
index f9bfaa56d..ba749ed99 100644
--- a/apps/ai-game-creator-shell/src/view/ui-editor/components/preview/PreviewWorkspace.tsx
+++ b/apps/ai-game-creator-shell/src/view/ui-editor/components/preview/PreviewWorkspace.tsx
@@ -56,7 +56,6 @@ export function PreviewWorkspace({
   );
   const [canvasSize, setCanvasSize] = useState({ width: 900, height: 640 });
   const canvasSizeRef = useRef(canvasSize);
-  canvasSizeRef.current = canvasSize;
   const [spaceHeld, setSpaceHeld] = useState(false);
   const [renderMode, setRenderMode] =
     useState<UiEditorRenderMode>('editor-overlay');
@@ -202,10 +201,12 @@ export function PreviewWorkspace({
     const element = viewportElementRef.current;
     if (!element) return;
     const updateSize = () => {
-      setCanvasSize({
+      const nextSize = {
         width: element.clientWidth || 900,
         height: element.clientHeight || 640,
-      });
+      };
+      canvasSizeRef.current = nextSize;
+      setCanvasSize(nextSize);
     };
     updateSize();
     const observer = new ResizeObserver(updateSize);