Fixing a code example