Correcting a typo in the code