product 2 change

Avoid shuffling the key (which is simply the partition index which is not used post shuffle).