Update references in the test files so tests work on production build