Fix prod.py