Fixing up the example