Bug fix from lesson 3.