diff --git a/x/multi-staking/module.go b/x/multi-staking/module.go index 955ca79c..17951d39 100644 --- a/x/multi-staking/module.go +++ b/x/multi-staking/module.go @@ -71,13 +71,11 @@ func (AppModuleBasic) GetTxCmd() *cobra.Command { } // GetQueryCmd returns the multi-staking and staking module's root query command. -func (AppModuleBasic) GetQueryCmd() (queryCmd *cobra.Command) { - queryCmd.AddCommand( - stakingcli.GetQueryCmd(), - cli.GetQueryCmd(), - ) +func (AppModuleBasic) GetQueryCmd() *cobra.Command { + cmd := stakingcli.GetQueryCmd() + cmd.AddCommand(cli.GetQueryCmd()) - return queryCmd + return cmd } // AppModule embeds the Cosmos SDK's x/staking AppModule where we only override @@ -138,6 +136,9 @@ func (am AppModule) LegacyQuerierHandler(legacyQuerierCdc *codec.LegacyAmino) sd func (am AppModule) RegisterServices(cfg module.Configurator) { stakingtypes.RegisterMsgServer(cfg.MsgServer(), multistakingkeeper.NewMsgServerImpl(am.keeper)) multistakingtypes.RegisterQueryServer(cfg.QueryServer(), multistakingkeeper.NewQueryServerImpl(am.keeper)) + + querier := stakingkeeper.Querier{Keeper: am.sk} + stakingtypes.RegisterQueryServer(cfg.QueryServer(), querier) } // InitGenesis initial genesis state for feeabs module