Fixed a bug in the example