Fixed typo in new code