Update test_audit_experiment.py

The tests need upgrading as well