Fix example code