Fixed error in example code.

Fixed an error in the example code.