fix bug in shuffle test