fix partitioner bug