Fix bug in example