Fix docstring after change