update with address.Codec

onomyprotocol · Jan 14, 2025 · 89094ee · 89094ee
1 parent b5245d9
commit 89094ee
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/x/psm/keeper/msg_server.go b/x/psm/keeper/msg_server.go
@@ -23,7 +23,7 @@ func NewMsgServerImpl(keeper Keeper) types.MsgServer {
 var _ types.MsgServer = msgServer{}
 
 func (k msgServer) UpdateParams(ctx context.Context, req *types.MsgUpdateParams) (*types.MsgUpdateParamsResponse, error) {
-	if _, err := k.keeper.addressCodec.StringToBytes(req.Authority); err != nil {
+	if _, err := k.keeper.AccountKeeper.AddressCodec().StringToBytes(req.Authority); err != nil {
 		return nil, errorsmod.Wrap(err, "invalid authority address")
 	}
 

diff --git a/x/psm/types/expected_keepers.go b/x/psm/types/expected_keepers.go
@@ -3,12 +3,14 @@ package types
 import (
 	"context"
 
+	addresscodec "cosmossdk.io/core/address"
 	"cosmossdk.io/math"
 	sdk "github.com/cosmos/cosmos-sdk/types"
 )
 
 // AccountKeeper defines the expected interface for the Account module.
 type AccountKeeper interface {
+	AddressCodec() addresscodec.Codec
 	GetAccount(context.Context, sdk.AccAddress) sdk.AccountI // only used for simulation
 	GetModuleAddress(name string) sdk.AccAddress
 	// Methods imported from account should be defined here