fixed a bug in the example