Fix docstring location