Address review comments
Address review comments

Signed-off-by: Rohit Nayak <83d5e1e49bd5f0ebbf6c9ba40416057fac1b5d76@planetscale.com>

mahak/vitess,mahak/vitess,vitessio/vitess,vitessio/vitess,vitessio/vitess,vitessio/vitess,vitessio/vitess,vitessio/vitess,mahak/vitess,mahak/vitess,mahak/vitess,vitessio/vitess,mahak/vitess,mahak/vitess,mahak/vitess,vitessio/vitess