Small AI Logic Fix