Fix number of partitions to follow example update