Fix the example code