fix minor bug in the example