diff --git a/tests/communication.rs b/tests/communication.rs
index d72566a..a326f86 100644
--- a/tests/communication.rs
+++ b/tests/communication.rs
@@ -1,3 +1,5 @@
+#![cfg(feature = "handshake")]
+
 use async_std::net::{TcpListener, TcpStream};
 use async_std::task;
 use async_tungstenite::{accept_async, client_async, WebSocketStream};
diff --git a/tests/handshakes.rs b/tests/handshakes.rs
index 3cd607c..ad9cd9a 100644
--- a/tests/handshakes.rs
+++ b/tests/handshakes.rs
@@ -1,3 +1,5 @@
+#![cfg(feature = "handshake")]
+
 use async_std::net::{TcpListener, TcpStream};
 use async_std::task;
 use async_tungstenite::{accept_async, client_async};