Fix audit.py