diff --git a/x/wasmplus/handler.go b/x/wasmplus/handler.go index fcfcc0f446..0b9bdfd571 100644 --- a/x/wasmplus/handler.go +++ b/x/wasmplus/handler.go @@ -4,7 +4,6 @@ import ( "strings" "github.com/gogo/protobuf/proto" - sdk "github.com/line/lbm-sdk/types" "github.com/line/wasmd/x/wasm" @@ -18,18 +17,21 @@ func NewHandler(k wasmtypes.ContractOpsKeeper) sdk.Handler { wasmHandler := wasm.NewHandler(k) return func(ctx sdk.Context, msg sdk.Msg) (*sdk.Result, error) { - var ( - res proto.Message - err error - ) - res, err = wasmHandler(ctx, msg) + res, err := wasmHandler(ctx, msg) if err != nil && strings.Contains(err.Error(), "MsgStoreCodeAndInstantiateContract") { // handle wasmplus service msg2, ok := msg.(*types.MsgStoreCodeAndInstantiateContract) if ok { + var ( + res proto.Message + err error + ) res, err = msgServer.StoreCodeAndInstantiateContract(sdk.WrapSDKContext(ctx), msg2) + return sdk.WrapServiceResult(ctx, res, err) + } else { + return nil, err } } - return sdk.WrapServiceResult(ctx, res, err) + return res, err } }