Simplify getPartition