Update test_audit_experiment.py

fix the test as well