Small fix on the example