diff --git a/Examples/xx-messenger/Sources/RestoreFeature/RestoreFeature.swift b/Examples/xx-messenger/Sources/RestoreFeature/RestoreFeature.swift
index 5339c89bbd3dba39edaf2cc8f0181e673b396f4b..00dbd841e6ffb17cf82f5ded723aa890fa0b69a8 100644
--- a/Examples/xx-messenger/Sources/RestoreFeature/RestoreFeature.swift
+++ b/Examples/xx-messenger/Sources/RestoreFeature/RestoreFeature.swift
@@ -122,6 +122,7 @@ public let restoreReducer = Reducer<RestoreState, RestoreAction, RestoreEnvironm
           backupData: backupData,
           backupPassphrase: backupPassphrase
         )
+        try env.messenger.listenForMessages()
         return .success(.finished)
       } catch {
         return .success(.failed(error as NSError))
diff --git a/Examples/xx-messenger/Tests/RestoreFeatureTests/RestoreFeatureTests.swift b/Examples/xx-messenger/Tests/RestoreFeatureTests/RestoreFeatureTests.swift
index 9a3cde3227357eb11d878c8c8f7fd7cc76f00fb5..16dac1475b1917ad9ff12d778c07989909d08e12 100644
--- a/Examples/xx-messenger/Tests/RestoreFeatureTests/RestoreFeatureTests.swift
+++ b/Examples/xx-messenger/Tests/RestoreFeatureTests/RestoreFeatureTests.swift
@@ -88,6 +88,7 @@ final class RestoreFeatureTests: XCTestCase {
 
     var didRestoreWithData: [Data] = []
     var didRestoreWithPassphrase: [String] = []
+    var didListenForMessages = 0
 
     let store = TestStore(
       initialState: RestoreState(
@@ -104,6 +105,9 @@ final class RestoreFeatureTests: XCTestCase {
       didRestoreWithPassphrase.append(passphrase)
       return restoreResult
     }
+    store.environment.messenger.listenForMessages.run = {
+      didListenForMessages += 1
+    }
 
     store.send(.set(\.$passphrase, backupPassphrase)) {
       $0.passphrase = backupPassphrase
@@ -115,6 +119,7 @@ final class RestoreFeatureTests: XCTestCase {
 
     XCTAssertNoDifference(didRestoreWithData, [backupData])
     XCTAssertNoDifference(didRestoreWithPassphrase, [backupPassphrase])
+    XCTAssertNoDifference(didListenForMessages, 1)
 
     store.receive(.finished) {
       $0.isRestoring = false