Fix bug with example