Updated the docstring there too.