fix error in return value