Correction to sample code.