fix the example code