Fix example to match doc