Fixing typo in return