Updated example code

Corrected small error in example code.