Fix docstring comment