Minor change to docstring