Minor change to sample code