diff --git a/codex-rs/core/src/tools/handlers/collab.rs b/codex-rs/core/src/tools/handlers/collab.rs
index 0f74ef658a..9e2f776a87 100644
--- a/codex-rs/core/src/tools/handlers/collab.rs
+++ b/codex-rs/core/src/tools/handlers/collab.rs
@@ -348,6 +348,7 @@ mod tests {
     use std::time::Duration;
     use tokio::sync::Mutex;
     use tokio::time::timeout;
+    use tokio_util::sync::CancellationToken;
 
     fn invocation(
         session: Arc<crate::codex::Session>,
@@ -362,6 +363,7 @@ mod tests {
             call_id: "call-1".to_string(),
             tool_name: tool_name.to_string(),
             payload,
+            cancellation_token: CancellationToken::new(),
         }
     }