Fix unit tests for production.