diff --git a/apps/ai-game-creator-shell/src-tauri/src/ui_editor/commands/separation/tree.rs b/apps/ai-game-creator-shell/src-tauri/src/ui_editor/commands/separation/tree.rs
index 8f9a454d0..d632796ef 100644
--- a/apps/ai-game-creator-shell/src-tauri/src/ui_editor/commands/separation/tree.rs
+++ b/apps/ai-game-creator-shell/src-tauri/src/ui_editor/commands/separation/tree.rs
@@ -100,9 +100,10 @@ pub fn construct_separation_state(state: &State) -> SeparationState {
             let size = image.pixel_size / ppu;
             let root_rect =
                 crate::ui_editor::layout::dimension::UIRect::new(nalgebra::Point2::origin(), size);
+            let root_resolved = tree.root.layout.transform.resolve(&root_rect);
             let mut children = Vec::new();
             for child in &tree.root.children {
-                collect_todo_nodes(child, &root_rect, ppu, &mut children);
+                collect_todo_nodes(child, &root_resolved, ppu, &mut children);
             }
             let root_extractable = is_unbound_image(&tree.root);
             if !root_extractable && children.is_empty() {