Update docstring to match code

- I'm guessing the code is the right one