diff --git a/CHANGELOG.md b/CHANGELOG.md
index d6d6937c7..4d871590f 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -49,6 +49,8 @@ Update btcstaking module docs to include EOI
 
 - [#324](https://github.com/babylonlabs-io/babylon/pull/324) Fix decrementing
 jailed fp counter
+- [#352](https://github.com/babylonlabs-io/babylon/pull/352) Fix: withdrawal cli 
+for rewards
 
 ### State Machine Breaking
 
diff --git a/x/incentive/client/cli/tx.go b/x/incentive/client/cli/tx.go
index 4a227a018..e61f58f84 100644
--- a/x/incentive/client/cli/tx.go
+++ b/x/incentive/client/cli/tx.go
@@ -2,6 +2,7 @@ package cli
 
 import (
 	"fmt"
+
 	sdk "github.com/cosmos/cosmos-sdk/types"
 	"github.com/spf13/cobra"
 
diff --git a/x/incentive/types/codec.go b/x/incentive/types/codec.go
index 6abb8fc7f..3d737ddd0 100644
--- a/x/incentive/types/codec.go
+++ b/x/incentive/types/codec.go
@@ -10,6 +10,7 @@ import (
 func RegisterCodec(cdc *codec.LegacyAmino) {
 	cdc.RegisterConcrete(&MsgWithdrawReward{}, "incentive/MsgWithdrawReward", nil)
 	cdc.RegisterConcrete(&MsgUpdateParams{}, "incentive/MsgUpdateParams", nil)
+	cdc.RegisterConcrete(&MsgSetWithdrawAddress{}, "incentive/MsgSetWithdrawAddress", nil)
 }
 
 func RegisterInterfaces(registry cdctypes.InterfaceRegistry) {
@@ -18,6 +19,7 @@ func RegisterInterfaces(registry cdctypes.InterfaceRegistry) {
 		(*sdk.Msg)(nil),
 		&MsgWithdrawReward{},
 		&MsgUpdateParams{},
+		&MsgSetWithdrawAddress{},
 	)
 
 	msgservice.RegisterMsgServiceDesc(registry, &_Msg_serviceDesc)
diff --git a/x/incentive/types/msg.go b/x/incentive/types/msg.go
index a0c4caffa..bd574efc9 100644
--- a/x/incentive/types/msg.go
+++ b/x/incentive/types/msg.go
@@ -8,4 +8,5 @@ import (
 var (
 	_ sdk.Msg = &MsgWithdrawReward{}
 	_ sdk.Msg = &MsgUpdateParams{}
+	_ sdk.Msg = &MsgSetWithdrawAddress{}
 )