Update doc comment to reflect API change