diff --git a/src/cowboy_websocket.erl b/src/cowboy_websocket.erl
index 992af526..1469c008 100644
--- a/src/cowboy_websocket.erl
+++ b/src/cowboy_websocket.erl
@@ -66,8 +66,8 @@
 -record(state, {
 	parent :: undefined | pid(),
 	ref :: ranch:ref(),
-	socket = undefined :: inet:socket() | undefined,
-	transport = undefined :: module(),
+	socket = undefined :: inet:socket() | {pid(), cowboy_stream:streamid()} | undefined,
+	transport = undefined :: module() | undefined,
 	handler :: module(),
 	key = undefined :: undefined | binary(),
 	timeout = infinity :: timeout(),