docstring fix