Fix docstring in wrong place