Minor fix to the sample