Fixed the code sample.

Fixed the code sample.