Update example to reflect API change