fix AI bugs