Fix example some more