fixed small issue in example code