Fixed wrong variable name in return