Fix prod mode