diff --git a/iOS/MasterFeed/Cell/MasterFeedTableViewSectionHeader.swift b/iOS/MasterFeed/Cell/MasterFeedTableViewSectionHeader.swift
index 901184ec6..07d6ab35c 100644
--- a/iOS/MasterFeed/Cell/MasterFeedTableViewSectionHeader.swift
+++ b/iOS/MasterFeed/Cell/MasterFeedTableViewSectionHeader.swift
@@ -29,7 +29,7 @@ class MasterFeedTableViewSectionHeader: UITableViewHeaderFooterView {
 		set {
 			if unreadCountView.unreadCount != newValue {
 				unreadCountView.unreadCount = newValue
-				unreadCountView.isHidden = (newValue < 1)
+				updateUnreadCountView()
 				setNeedsLayout()
 			}
 		}
@@ -51,6 +51,7 @@ class MasterFeedTableViewSectionHeader: UITableViewHeaderFooterView {
 	var disclosureExpanded = false {
 		didSet {
 			updateDisclosureImage()
+			updateUnreadCountView()
 		}
 	}
 	
@@ -141,6 +142,14 @@ private extension MasterFeedTableViewSectionHeader {
 			}
 		}
 	}
+	
+	func updateUnreadCountView() {
+		if !disclosureExpanded && unreadCount > 0 {
+			unreadCountView.isHidden = false
+		} else {
+			self.unreadCountView.isHidden = true
+		}
+	}
 
 	func addSubviewAtInit(_ view: UIView) {
 		addSubview(view)
diff --git a/iOS/MasterFeed/MasterFeedViewController.swift b/iOS/MasterFeed/MasterFeedViewController.swift
index 65e6092a2..039e5b036 100644
--- a/iOS/MasterFeed/MasterFeedViewController.swift
+++ b/iOS/MasterFeed/MasterFeedViewController.swift
@@ -11,7 +11,6 @@ import Account
 import Articles
 import RSCore
 import RSTree
-import SwiftUI
 
 class MasterFeedViewController: UITableViewController, UndoableCommandRunner {