From f2a0dda03aacaeebae6d6af94206d1790ce2baa3 Mon Sep 17 00:00:00 2001
From: Alex Koshelev <koshelev@meta.com>
Date: Tue, 31 Oct 2023 00:03:28 -0700
Subject: [PATCH] Fix shuttle tests

---
 src/helpers/gateway/mod.rs | 2 ++
 1 file changed, 2 insertions(+)

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};