Fix bug in the example