Fixing code example