Small AI fixes