fix the example code at the comment