Fix example from last commit