Fix the production build