Fix sub exercise 3. Thanks @metahys.