Fix type error on Exercise 2