Correction in sample code

Correction in sample code