Fix running on production