Minor fix to example