Fix reference in docstring