Correct doc string to match the docs