Fix code example and lint issue