Bugfix on example code