Fix the sample code