fix error in return type