Fix error in Question 1