Fix wrong answer for Lab 20