Exercise 4 - Fix spec