1 files changed, 6 insertions, 4 deletions
diff --git a/src/message/handshake/init.rs b/src/message/handshake/init.rs
index df1b29e..51514ff 100644
--- a/src/message/handshake/init.rs
+++ b/src/message/handshake/init.rs
@@ -15,12 +15,14 @@ impl Init {
         }
     }
 
-    pub fn compression(mut self, v: bool) {
-        self.compression = v
+    pub fn compression(mut self, v: bool) -> Self {
+        self.compression = v;
+        self
     }
 
-    pub fn tls(mut self, v: bool) {
-        self.tls = v
+    pub fn tls(mut self, v: bool) -> Self {
+        self.tls = v;
+        self
     }
 
     pub fn serialize(self) -> Vec<u8> {