DOC: Fixed docstring