Correcting a small bug in the example