Fix a function in doc