fix the exercise