We need the is_correct values for supervision