diff --git a/src/handshake/machine.rs b/src/handshake/machine.rs
index cb04144..7348cf7 100644
--- a/src/handshake/machine.rs
+++ b/src/handshake/machine.rs
@@ -1,3 +1,5 @@
+//! handshake machine.
+
 use bytes::Buf;
 use log::*;
 use std::io::{Cursor, Read, Write};
diff --git a/src/handshake/mod.rs b/src/handshake/mod.rs
index 00eeb74..9310e04 100644
--- a/src/handshake/mod.rs
+++ b/src/handshake/mod.rs
@@ -3,8 +3,6 @@
 pub mod client;
 pub mod headers;
 pub mod server;
-
-#[allow(missing_docs)]
 pub mod machine;
 
 use std::{
diff --git a/src/handshake/server.rs b/src/handshake/server.rs
index d280ffa..9a7a20c 100644
--- a/src/handshake/server.rs
+++ b/src/handshake/server.rs
@@ -93,8 +93,7 @@ pub fn create_response_with_body<T>(
     Ok(create_parts(request)?.body(generate_body())?)
 }
 
-// Assumes that this is a valid response
-#[allow(missing_docs)]
+/// Write response to the stream `w`.
 pub fn write_response<T>(mut w: impl io::Write, response: &HttpResponse<T>) -> Result<()> {
     writeln!(
         w,