diff --git a/crates/agent_ui/src/agent_ui.rs b/crates/agent_ui/src/agent_ui.rs
index a57e5b0563..946d24d9ae 100644
--- a/crates/agent_ui/src/agent_ui.rs
+++ b/crates/agent_ui/src/agent_ui.rs
@@ -9,7 +9,6 @@ mod context_picker;
 mod context_server_configuration;
 mod context_strip;
 mod debug;
-mod history_store;
 mod inline_assistant;
 mod inline_prompt_editor;
 mod language_model_selector;
@@ -22,6 +21,7 @@ mod terminal_codegen;
 mod terminal_inline_assistant;
 mod text_thread_editor;
 mod thread_history;
+mod thread_history2;
 mod tool_compatibility;
 mod ui;
 
diff --git a/crates/agent_ui/src/thread_history.rs b/crates/agent_ui/src/thread_history.rs
index 04ccec0975..b8d1db88d6 100644
--- a/crates/agent_ui/src/thread_history.rs
+++ b/crates/agent_ui/src/thread_history.rs
@@ -1,5 +1,5 @@
-use crate::history_store::{HistoryEntry, HistoryStore};
 use crate::{AgentPanel, RemoveSelectedThread};
+use agent::history_store::{HistoryEntry, HistoryStore};
 use chrono::{Datelike as _, Local, NaiveDate, TimeDelta};
 use editor::{Editor, EditorEvent};
 use fuzzy::{StringMatch, StringMatchCandidate};