Minor fix to code examples