diff --git a/codex-rs/core/tests/suite/model_runtime_selectors.rs b/codex-rs/core/tests/suite/model_runtime_selectors.rs index 029df85fe4..5a0823c27a 100644 --- a/codex-rs/core/tests/suite/model_runtime_selectors.rs +++ b/codex-rs/core/tests/suite/model_runtime_selectors.rs @@ -397,8 +397,16 @@ async fn guardian_stays_disabled_when_model_selects_multi_agent_v2() -> Result<( } test.codex.submit(Op::Interrupt).await?; - let root_request = root_mock.single_request(); - let guardian_request = guardian_mock.single_request(); + let root_request = root_mock + .requests() + .into_iter() + .next() + .expect("expected root request"); + let guardian_request = guardian_mock + .requests() + .into_iter() + .next() + .expect("expected guardian request"); assert_eq!( ( models_mock.requests().len(),