Fixed up the sample code.