fixed the ai