diff --git a/src/mail.js b/src/mail.js
index 66bd1449e..be2e7e9a4 100644
--- a/src/mail.js
+++ b/src/mail.js
@@ -10,6 +10,8 @@ exports = module.exports = {
     del: del,
     update: update,
 
+    addDnsRecords: addDnsRecords,
+
     setMailFromValidation: setMailFromValidation,
     setCatchAllAddress: setCatchAllAddress,
     setMailRelay: setMailRelay,
diff --git a/src/setup.js b/src/setup.js
index 78b5481c1..826b684e6 100644
--- a/src/setup.js
+++ b/src/setup.js
@@ -296,7 +296,7 @@ function restore(backupConfig, backupId, version, callback) {
                 autoprovision,
                 // currently, our suggested restore flow is after a dnsSetup. This re-creates DKIM keys and updates the DNS
                 // for this reason, we have to re-setup DNS after a restore. Once we have a 100% IP based restore, we can skip this
-                mail.update.bind(null, config.adminDomain()),
+                mail.addDnsRecords.bind(null, config.adminDomain()),
                 shell.sudo.bind(null, 'restart', [ RESTART_CMD ])
             ], function (error) {
                 debug('restore:', error);