Fix docstring reference