Fix broken AI