fix NA return bug