Correct the docstring.

Correct the docstring.