fixed a problem with the docstring