Changed docstring to refer to the correct function