minor change to the ai