fix ai to new proto