updated ai tests