Fixed small bug in example code