Add support for overriding shard size for a time