Chapter 1, make test pass with smallest change