Fix example code in comment