chore: address PR comments

app/app_test.go

@@ -109,8 +109,18 @@ func TestStakingRouterDisabled(t *testing.T) {
 		AppOpts:            TmpAppOptions(),
 	})
 
-	msg := &stktypes.MsgCreateValidator{}
+	msgs := []sdk.Msg{
+		&stktypes.MsgCreateValidator{},
+		&stktypes.MsgBeginRedelegate{},
+		&stktypes.MsgCancelUnbondingDelegation{},
+		&stktypes.MsgDelegate{},
+		&stktypes.MsgEditValidator{},
+		&stktypes.MsgUndelegate{},
+		&stktypes.MsgUpdateParams{},
+	}
 
-	msgHandler := app.MsgServiceRouter().HandlerByTypeURL(sdk.MsgTypeURL(msg))
-	require.Nil(t, msgHandler)
+	for _, msg := range msgs {
+		msgHandler := app.MsgServiceRouter().HandlerByTypeURL(sdk.MsgTypeURL(msg))
+		require.Nil(t, msgHandler)
+	}
 }

x/epoching/README.md

@@ -147,6 +147,13 @@ to a node it will error out. Since there is no message to be received
 by the Staking module, all of commands under `babylond tx staking` were
 removed.
 
+**Staking module Migrations.** Since the staking module msg server
+was not registered in the app routes, the function of staking `AppModule`
+that register the migrations and the query server are in the
+[`app.go`](https://github.com/babylonlabs-io/babylon/blob/cf6c0e3873133331d95267a58da06c04a3e2c601/app/app.go#L608). If a new
+version of the cosmos-sdk is released and there is a new migration,
+it is needed to register the migration there too.
+
 **Delaying wrapped messages to the end of epochs.** The Epoching module
 maintains a message queue for each epoch. Upon each wrapped message, the
 Epoching module performs basic sanity checks, then enqueues the message to the