fixup: integrate bg router

pokt-network · Jun 22, 2023 · ca0b962 · ca0b962
1 parent bf6a85d
commit ca0b962
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/p2p/module.go b/p2p/module.go
@@ -245,7 +245,16 @@ func (m *p2pModule) Send(addr cryptoPocket.Address, msg *anypb.Any) error {
 	}
 
 	unstakedSendErr := m.unstakedActorRouter.Send(poktEnvelopeBz, addr)
-	stakedSendErr := m.stakedActorRouter.Send(poktEnvelopeBz, addr)
+
+	isStaked, err := m.isStakedActor()
+	if err != nil {
+		return err
+	}
+
+	var stakedSendErr error
+	if isStaked {
+		stakedSendErr = m.stakedActorRouter.Send(poktEnvelopeBz, addr)
+	}
 
 	if errors.Is(unstakedSendErr, typesP2P.ErrUnknownPeer) &&
 		errors.Is(stakedSendErr, typesP2P.ErrUnknownPeer) {