Corrected typo and added return statement