Fix globe icon not disappearing

duckduckgo · Jan 23, 2025 · 5323ec5 · 5323ec5
1 parent b20b0a3
commit 5323ec5
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/DuckDuckGo/SpecialErrorPage/SpecialErrorPageNavigationHandler.swift b/DuckDuckGo/SpecialErrorPage/SpecialErrorPageNavigationHandler.swift
@@ -66,12 +66,7 @@ extension SpecialErrorPageNavigationHandler: WebViewNavigationHandling {
 
     @MainActor
     func handleDecidePolicy(for navigationAction: WKNavigationAction, webView: WKWebView) {
-        // A new main navigation is starting reset the error
-        if navigationAction.isTargetingMainFrame() {
-            errorData = nil
-            failedURL = nil
-            isSpecialErrorPageVisible = false
-        }
+        guard navigationAction.isTargetingMainFrame() else { return }
         maliciousSiteProtectionNavigationHandler.makeMaliciousSiteDetectionTask(for: navigationAction, webView: webView)
     }
 
@@ -102,6 +97,7 @@ extension SpecialErrorPageNavigationHandler: WebViewNavigationHandling {
             return true
         case .navigationNotHandled:
             isSpecialErrorPageRequest = false
+            isSpecialErrorPageVisible = false
             return false
         }
     }