Please update the docstring to reflect the new param.