diff --git a/crates/engine/src/cmd/buildcmds/shape_builder.rs b/crates/engine/src/cmd/buildcmds/shape_builder.rs
index b230dd427..7d9b7e7ab 100644
--- a/crates/engine/src/cmd/buildcmds/shape_builder.rs
+++ b/crates/engine/src/cmd/buildcmds/shape_builder.rs
@@ -1,7 +1,7 @@
 use super::{
-    Beside, BuilderMap, EngineError, FillRule, Node, NodeId, PathData, Point, Rect, Result, Session, SubPath, Touched, Value, b, bad, check_numbers,
-    f64_or, fill_param, finite, gap_param, ids_json, json, marquee_param, po, points_param, remnant, replace_roots, replace_with_pieces, root_ids,
-    shape_node, switch_or,
+    Arc, Beside, BuilderMap, EngineError, FillRule, Node, NodeId, PathData, Point, Rect, Result, Session, SubPath, Touched, Value, b, bad,
+    check_numbers, f64_or, fill_param, finite, gap_param, ids_json, json, marquee_param, po, points_param, remnant, replace_roots,
+    replace_with_pieces, root_ids, shape_node, switch_or,
 };
 
 pub(super) fn sb_merge(s: &mut Session, p: &Value) -> Result<Value> {
diff --git a/crates/engine/src/session.rs b/crates/engine/src/session.rs
index 1eaf15b88..340c081a2 100644
--- a/crates/engine/src/session.rs
+++ b/crates/engine/src/session.rs
@@ -1,4 +1,4 @@
-use super::command_sets::{SELECTS_GIVEN_IDS, given_ids, strip_replay_only};
+use super::command_sets::{SELECTS_GIVEN_IDS, given_ids, strip_replay_only, waits_for_tool_edit};
 use super::doc_state::DocState;
 use super::error::{EngineError, Result};
 use super::history::NamedStep;
diff --git a/crates/engine/src/session/execute.rs b/crates/engine/src/session/execute.rs
index 0a5a31b90..373e59f5c 100644
--- a/crates/engine/src/session/execute.rs
+++ b/crates/engine/src/session/execute.rs
@@ -1,5 +1,6 @@
 use super::{
     CommandSpec, EngineError, NodeId, Result, SELECTS_GIVEN_IDS, Selection, Session, Value, cmd, find_command, given_ids, guard, strip_replay_only,
+    waits_for_tool_edit,
 };
 
 impl Session {