Update documentation to reflect the argument change