Fix: no longer assume the index of the first element of a partition is a multiple of the partition size