Fix sharding test