diff --git a/protocol/base/base.go b/protocol/base/base.go
index a3e0358..98468c6 100644
--- a/protocol/base/base.go
+++ b/protocol/base/base.go
@@ -9,6 +9,7 @@ import (
 type IncomingMessage interface {
 	HandleMessage(node interfaces.Node, peer net.Peer, verifyId bool) error
 	SetIncomingMessage(msg IncomingMessage)
+	IncomingMessage() IncomingMessage
 	Self() IncomingMessage
 	SetSelf(self IncomingMessage)
 	msgpack.CustomDecoder
diff --git a/protocol/base/incoming_message.go b/protocol/base/incoming_message.go
index 669f0c1..d481d14 100644
--- a/protocol/base/incoming_message.go
+++ b/protocol/base/incoming_message.go
@@ -22,6 +22,7 @@ type IncomingMessageImpl struct {
 	original []byte
 	known    bool
 	self     IncomingMessage
+	incoming IncomingMessage
 }
 
 func (i *IncomingMessageImpl) Self() IncomingMessage {
@@ -53,13 +54,12 @@ func (i *IncomingMessageImpl) Original() []byte {
 }
 
 func (i *IncomingMessageImpl) SetIncomingMessage(msg IncomingMessage) {
-	if msgImpl, ok := msg.(*IncomingMessageImpl); ok {
-		*i = *msgImpl
-		i.known = true
-	} else {
-		// Handle the error or panic
-		panic("msg is not of type *IncomingMessageImpl")
-	}
+	i.incoming = msg
+	i.known = true
+}
+
+func (i *IncomingMessageImpl) IncomingMessage() IncomingMessage {
+	return i.incoming
 }
 
 func (i *IncomingMessageImpl) GetKind() types.ProtocolMethod {