diff --git a/apps/ai-game-creator-shell/src-tauri/src/ui_editor/commands/separation/prompt/extract.rs b/apps/ai-game-creator-shell/src-tauri/src/ui_editor/commands/separation/prompt/extract.rs
index 8a40dbaf7..766363f18 100644
--- a/apps/ai-game-creator-shell/src-tauri/src/ui_editor/commands/separation/prompt/extract.rs
+++ b/apps/ai-game-creator-shell/src-tauri/src/ui_editor/commands/separation/prompt/extract.rs
@@ -1,6 +1,8 @@
 use crate::ui_editor::commands::separation::model::{SeparationState, SeparationTree};
-use crate::ui_editor::commands::separation::workflow::batch::terminal_node_ids;
-use crate::ui_editor::commands::separation::{sanitize_prompt_text, SeparationNode, SeparationNodeKind, MAX_NOTE_DESCRIPTION_CHARS, MAX_REWORK_NOTE_CHARS};
+use crate::ui_editor::commands::separation::{
+    sanitize_prompt_text, SeparationNode, SeparationNodeKind, MAX_NOTE_DESCRIPTION_CHARS,
+    MAX_REWORK_NOTE_CHARS,
+};
 use crate::ui_editor::utils::NodeId;
 use serde::Serialize;
 use std::collections::HashSet;
@@ -63,7 +65,11 @@ pub(crate) fn gen_extract_prompt(
         .iter()
         .map(|node| node.id.clone())
         .collect::<HashSet<_>>();
-    let terminal_ids = terminal_node_ids(state);
+    let terminal_ids = state
+        .bound
+        .iter()
+        .map(|node| node.node_id.clone())
+        .collect::<HashSet<_>>();
     let mut index = 1;
     let document = ExtractPromptDocument {
         ui_layer_tree: project_node(&tree.root, &target_ids, &terminal_ids, &mut index),