Fix partitioner bug