DOC: tweak docstring