diff --git a/client/socket.go b/client/socket.go
index 1d47588..367ae72 100644
--- a/client/socket.go
+++ b/client/socket.go
@@ -13,6 +13,8 @@ import (
 )
 
 type Socket interface {
+	Context() SocketContext
+
 	// Subprotocol returns the negotiated protocol for the connection.
 	Subprotocol() string