Fixing small typo in the code