Fix issue with exercise 1