diff --git a/internal/client/websocket.go b/internal/client/websocket.go
index 3e4cf05..e1b5ce7 100644
--- a/internal/client/websocket.go
+++ b/internal/client/websocket.go
@@ -62,3 +62,10 @@ func (ws *WSOverTLS) Handshake(rawConn net.Conn, authInfo AuthInfo) (sessionKey
 
 	return
 }
+
+func (ws *WSOverTLS) Close() error {
+	if ws.WebSocketConn != nil {
+		return ws.WebSocketConn.Close()
+	}
+	return nil
+}