Fix error on lesson 3