Fix docstring

I changed my mind in the middle and forgot to change the docstring