Fixed the docstring