diff --git a/tests/auto-run-fresh-attempt-reset.test.js b/tests/auto-run-fresh-attempt-reset.test.js
index bb0bc0d..3c8622a 100644
--- a/tests/auto-run-fresh-attempt-reset.test.js
+++ b/tests/auto-run-fresh-attempt-reset.test.js
@@ -67,6 +67,7 @@ const bundle = [
   extractFunction('getAutoRunRoundRetryCount'),
   extractFunction('formatAutoRunFailureReasons'),
   extractFunction('logAutoRunFinalSummary'),
+  extractFunction('waitBetweenAutoRunRounds'),
   extractFunction('autoRunLoop'),
 ].join('\n');
 
diff --git a/tests/step9-localhost-cleanup-scope.test.js b/tests/step9-localhost-cleanup-scope.test.js
index 4bf1a99..ffb9670 100644
--- a/tests/step9-localhost-cleanup-scope.test.js
+++ b/tests/step9-localhost-cleanup-scope.test.js
@@ -55,6 +55,7 @@ const bundle = [
   extractFunction('normalizeEmailGenerator'),
   extractFunction('parseUrlSafely'),
   extractFunction('isHotmailProvider'),
+  extractFunction('isCustomMailProvider'),
   extractFunction('isGeneratedAliasProvider'),
   extractFunction('shouldUseCustomRegistrationEmail'),
   extractFunction('isLocalhostOAuthCallbackUrl'),