Update docs with renamed callback argument