Fix docstring error