fix to exercise 8