fix sharding issue

fix sharding issue