Small AI fix