Fix a type with class 2 exercise 1