Bug fix for partitioner.