diff --git a/codex-rs/codex-api/src/endpoint/session_tests.rs b/codex-rs/codex-api/src/endpoint/session_tests.rs
index 82cf9c0748..d420b4b498 100644
--- a/codex-rs/codex-api/src/endpoint/session_tests.rs
+++ b/codex-rs/codex-api/src/endpoint/session_tests.rs
@@ -1,4 +1,5 @@
 use super::*;
+use pretty_assertions::assert_eq;
 use std::sync::Mutex as StdMutex;
 
 #[derive(Default)]