Correct the example code