Fix docstring issue