Fixed the challenge return value