Fix prod build