Fix example to reflect latest changes