AI bug fix 2