correction in return value