Fix error with example code