1 files changed, 7 insertions, 4 deletions
diff --git a/test/unit/idStore-migration-test.js b/test/unit/idStore-migration-test.js
index 8532ac914..ad4ce2096 100644
--- a/test/unit/idStore-migration-test.js
+++ b/test/unit/idStore-migration-test.js
@@ -79,11 +79,14 @@ describe('IdentityStore to KeyringController migration', function() {
   })
 
   describe('entering a password', function() {
-    it('should identify an old wallet as an initialized keyring', function() {
+    it('should identify an old wallet as an initialized keyring', function(done) {
       keyringController.configManager.setWallet('something')
-      const state = keyringController.getState()
-      assert(state.isInitialized, 'old vault counted as initialized.')
-      assert(!state.lostAccounts, 'no lost accounts')
+      keyringController.getState()
+      .then((state) => {
+        assert(state.isInitialized, 'old vault counted as initialized.')
+        assert(!state.lostAccounts, 'no lost accounts')
+        done()
+      })
     })
   })
 })