Set correct partitions