Update docstring similarly here.