Fix test after Shuffler changes.