Fix example to be consistent with the docs.