small AI fix