Fixing typo

Looks like there is a typo in the code.