fix AI bug