Fixed a small error in the example