Fix code example