diff --git a/aquatic_ws/src/lib/network/connection.rs b/aquatic_ws/src/lib/network/connection.rs
index 1404e3c..acb99a6 100644
--- a/aquatic_ws/src/lib/network/connection.rs
+++ b/aquatic_ws/src/lib/network/connection.rs
@@ -229,7 +229,7 @@ impl Connection {
     ) -> Self {
         Self {
             valid_until,
-            inner: Either::Right(HandshakeMachine::TcpStream(tcp_stream))
+            inner: Either::Right(HandshakeMachine::new(tcp_stream))
         }
     }