Sorry I forget to update doc-string.
Here's the correct patch.