diff --git a/protocol/handshake_open.go b/protocol/handshake_open.go
index 38667fb..0f86684 100644
--- a/protocol/handshake_open.go
+++ b/protocol/handshake_open.go
@@ -22,6 +22,10 @@ type HandshakeOpen struct {
 	base.IncomingMessageHandler
 }
 
+func (h *HandshakeOpen) SetHandshake(handshake []byte) {
+	h.handshake = handshake
+}
+
 func (h HandshakeOpen) Challenge() []byte {
 	return h.challenge
 }