diff --git a/apps/ai-game-creator-shell/src-tauri/src/ui_editor/commands/separation/mod.rs b/apps/ai-game-creator-shell/src-tauri/src/ui_editor/commands/separation/mod.rs
index d103da6e1..edef9453e 100644
--- a/apps/ai-game-creator-shell/src-tauri/src/ui_editor/commands/separation/mod.rs
+++ b/apps/ai-game-creator-shell/src-tauri/src/ui_editor/commands/separation/mod.rs
@@ -7,10 +7,16 @@ mod workflow;
 pub use model::*;
 pub use persistence::*;
 pub use tree::*;
+pub use workflow::apply_batch_patch;
 pub(crate) use workflow::separate_ui_impl;
 #[cfg(test)]
 mod tests {
     use super::*;
+    use crate::ui_editor::component::Component;
+    use crate::ui_editor::layout::node::Node;
+    use crate::ui_editor::state::{State, UITree};
+    use crate::ui_editor::utils::{NodeId, UIDesignImageId};
+    use std::path::Path;
     use crate::ui_editor::component::image::{ImageComponent, ImageType};
     use crate::ui_editor::layout::children_display_mode::ChildrenDisplayMode;
     use crate::ui_editor::layout::control_layout::ControlLayout;