Fix misplaced docstring