fix tests on production