Enable polling packet tunnel even in blocked state

1 files changed, 3 insertions, 5 deletions

diff --git a/ios/MullvadVPN/TunnelManager/TunnelManager.swift b/ios/MullvadVPN/TunnelManager/TunnelManager.swift
index 682607a68f..3ca7ccb31a 100644
--- a/ios/MullvadVPN/TunnelManager/TunnelManager.swift
+++ b/ios/MullvadVPN/TunnelManager/TunnelManager.swift
@@ -736,11 +736,11 @@ final class TunnelManager: StorePaymentObserver, @unchecked Sendable {
             // while the tunnel process is trying to connect.
             startPollingTunnelStatus(interval: establishingTunnelStatusPollInterval)
 
-        case .connected, .waitingForConnectivity(.noConnection):
+        case .connected, .waitingForConnectivity(.noConnection), .waitingForConnectivity(.noNetwork):
             // Start polling tunnel status to keep connectivity status up to date.
             startPollingTunnelStatus(interval: establishedTunnelStatusPollInterval)
 
-        case .pendingReconnect, .disconnecting, .disconnected, .waitingForConnectivity(.noNetwork):
+        case .pendingReconnect, .disconnecting, .disconnected:
             // Stop polling tunnel status once connection moved to final state.
             cancelPollingTunnelStatus()
 
@@ -748,12 +748,10 @@ final class TunnelManager: StorePaymentObserver, @unchecked Sendable {
             switch blockedStateReason {
             case .deviceRevoked, .invalidAccount:
                 handleBlockedState(reason: blockedStateReason)
+                cancelPollingTunnelStatus()
             default:
                 break
             }
-
-            // Stop polling tunnel status once blocked state has been determined.
-            cancelPollingTunnelStatus()
         }
 
         DispatchQueue.main.async {