fix 1 logic error in exercise 2.4