Fix docstring