AI fixes