fix partition_by example output