fixes jai tests