Fixed a small error in example code