fix docstring