Fixing sloppy example code