Fix index error in ShuffleMode

Fix index error in ShuffleMode