DOC: fix docstring