From 8015f6e254bfcf2bf4ff37086514d43396917ccf Mon Sep 17 00:00:00 2001
From: William Banfield <wbanfield@gmail.com>
Date: Thu, 20 Oct 2022 16:59:06 -0400
Subject: [PATCH] use correct type in wrap

---
 p2p/types.go                        | 2 ++
 proto/tendermint/mempool/message.go | 6 +++---
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/p2p/types.go b/p2p/types.go
index e8182a6f3..70a5ee536 100644
--- a/p2p/types.go
+++ b/p2p/types.go
@@ -27,6 +27,8 @@ type Unwrapper interface {
 }
 
 type Wrapper interface {
+	proto.Message
+
 	// Wrap will take the underlying message and wrap it in its wrapper type.
 	Wrap() (proto.Message, error)
 }
diff --git a/proto/tendermint/mempool/message.go b/proto/tendermint/mempool/message.go
index 4686aa6f2..0e9d797f8 100644
--- a/proto/tendermint/mempool/message.go
+++ b/proto/tendermint/mempool/message.go
@@ -7,12 +7,12 @@ import (
 	"github.com/tendermint/tendermint/p2p"
 )
 
-var _ p2p.Wrapper = &Message_Txs{}
+var _ p2p.Wrapper = &Txs{}
 
 // Wrap implements the p2p Wrapper interface and wraps a mempool message.
-func (m *Message_Txs) Wrap() (proto.Message, error) {
+func (m *Txs) Wrap() (proto.Message, error) {
 	return &Message{
-		Sum: m,
+		Sum: &Message_Txs{Txs: m},
 	}, nil
 }