Fix a NetP UI presentation issue.

duckduckgo · Nov 25, 2023 · 4479650 · 4479650
1 parent 67f2437
commit 4479650
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/DuckDuckGo/VPNWaitlistViewController.swift b/DuckDuckGo/VPNWaitlistViewController.swift
@@ -122,9 +122,13 @@ extension VPNWaitlistViewController: WaitlistViewModelDelegate {
     func waitlistViewModel(_ viewModel: WaitlistViewModel, didTriggerCustomAction action: WaitlistViewModel.ViewCustomAction) {
         if action == .openNetworkProtectionInviteCodeScreen {
             let networkProtectionViewController = NetworkProtectionRootViewController { [weak self] in
-                self?.navigationController?.popToRootViewController(animated: true)
-                let newRootViewController = NetworkProtectionRootViewController()
-                self?.navigationController?.pushViewController(newRootViewController, animated: true)
+                guard let self = self, let rootViewController = self.navigationController?.viewControllers.first else {
+                    assertionFailure("Failed to show NetP status view")
+                    return
+                }
+
+                let networkProtectionRootViewController = NetworkProtectionRootViewController()
+                self.navigationController?.setViewControllers([rootViewController, networkProtectionRootViewController], animated: true)
             }
 
             self.navigationController?.pushViewController(networkProtectionViewController, animated: true)