Fixed a bug in the partitioner