Simplify partialShuffle