Small sample code tweak