Fix bug for example