Small fix to example