Fix the docstring