diff --git a/DuckDuckGo/OmniBar.swift b/DuckDuckGo/OmniBar.swift index 01415d572d..40f2edc04c 100644 --- a/DuckDuckGo/OmniBar.swift +++ b/DuckDuckGo/OmniBar.swift @@ -29,6 +29,7 @@ extension OmniBar: NibLoading {} public enum OmniBarIcon: String { case duckPlayer = "DuckPlayerURLIcon" + case specialError = "Globe-24" } class OmniBar: UIView { @@ -300,10 +301,15 @@ class OmniBar: UIView { showCustomIcon(icon: .duckPlayer) return } - - privacyInfoContainer.privacyIcon.isHidden = privacyInfo.isSpecialErrorPageVisible + + if privacyInfo.isSpecialErrorPageVisible { + showCustomIcon(icon: .specialError) + return + } + let icon = PrivacyIconLogic.privacyIcon(for: privacyInfo) privacyInfoContainer.privacyIcon.updateIcon(icon) + privacyInfoContainer.privacyIcon.isHidden = false customIconView.isHidden = true }