Final AI fixes