fix the exercise 5.4