Docstring fix