diff --git a/scripts/migrate-imports.js b/scripts/migrate-imports.js
index 63acce83c..5e9dcc44c 100755
--- a/scripts/migrate-imports.js
+++ b/scripts/migrate-imports.js
@@ -163,6 +163,7 @@ function getUpgradeablePath (file) {
 module.exports = {
   pathUpdates,
   updateImportPaths,
+  getUpgradeablePath,
 };
 
 if (require.main === module) {
diff --git a/test/migrate-imports.test.js b/test/migrate-imports.test.js
index a85fce21e..0872033d3 100644
--- a/test/migrate-imports.test.js
+++ b/test/migrate-imports.test.js
@@ -2,12 +2,16 @@ const path = require('path');
 const { promises: fs, constants: { F_OK } } = require('fs');
 const { expect } = require('chai');
 
-const { pathUpdates, updateImportPaths } = require('../scripts/migrate-imports.js');
+const { pathUpdates, updateImportPaths, getUpgradeablePath } = require('../scripts/migrate-imports.js');
 
 describe('migrate-imports.js', function () {
   it('every new path exists', async function () {
     for (const p of Object.values(pathUpdates)) {
-      await fs.access(path.join('contracts', p), F_OK);
+      try {
+        await fs.access(path.join('contracts', p), F_OK);
+      } catch (e) {
+        await fs.access(path.join('contracts', getUpgradeablePath), F_OK);
+      }
     }
   });