diff --git a/src/helpers/gateway/mod.rs b/src/helpers/gateway/mod.rs
index 048e9d119..7f4bb36f6 100644
--- a/src/helpers/gateway/mod.rs
+++ b/src/helpers/gateway/mod.rs
@@ -12,6 +12,8 @@ use std::num::NonZeroUsize;
 
 pub(super) use receive::ReceivingEnd;
 pub(super) use send::SendingEnd;
+#[cfg(all(test, feature = "shuttle"))]
+use shuttle::future as tokio;
 #[cfg(feature = "stall-detection")]
 pub(super) use {observable::Observed, stall_detection::InstrumentedGateway};