Fix AI breakage