Fix the partitioner