Making tests work on production build