diff --git a/types/v1/tx.go b/types/v1/tx.go index d8c9699..eade1a2 100644 --- a/types/v1/tx.go +++ b/types/v1/tx.go @@ -259,10 +259,8 @@ func (t *Tx) GetAddresses() []string { return append(addresses, t.From, t.To) case TxContractCall: return append(addresses, t.From, t.To) - case TxStakeDelegate, TxStakeRedelegate: + case TxStakeDelegate, TxStakeRedelegate, TxStakeUndelegate, TxStakeClaimRewards: return append(addresses, t.From) - case TxStakeUndelegate, TxStakeClaimRewards: - return append(addresses, t.To) default: return addresses } diff --git a/types/v1/tx_test.go b/types/v1/tx_test.go index e9a48a6..37b6649 100644 --- a/types/v1/tx_test.go +++ b/types/v1/tx_test.go @@ -154,7 +154,17 @@ func TestTx_GetAddresses(t *testing.T) { To: "to", Metadata: &Transfer{}, }, - expected: []string{"to"}, + expected: []string{"from"}, + }, + { + name: "claim_rewards", + tx: Tx{ + Type: TxStakeClaimRewards, + From: "from", + To: "to", + Metadata: &Transfer{}, + }, + expected: []string{"from"}, }, { name: "contract_call",