diff --git a/axum/src/extract/ws.rs b/axum/src/extract/ws.rs
index 3fc807bfc8..f44ee07f86 100644
--- a/axum/src/extract/ws.rs
+++ b/axum/src/extract/ws.rs
@@ -845,6 +845,7 @@ mod tests {
 
     use super::*;
     use crate::{body::Body, routing::get, test_helpers::spawn_service, Router};
+    use futures_util::sink::SinkExt;
     use http::{Request, Version};
     use tokio_tungstenite::tungstenite;
     use tower::ServiceExt;