Fix incorrect docstring