Add an assert to prevent wrong predictions