fix ai problem