diff --git a/iOS/MasterFeed/MasterFeedViewController.swift b/iOS/MasterFeed/MasterFeedViewController.swift index 20f92c253..d7232e61c 100644 --- a/iOS/MasterFeed/MasterFeedViewController.swift +++ b/iOS/MasterFeed/MasterFeedViewController.swift @@ -99,10 +99,8 @@ class MasterFeedViewController: UITableViewController, UndoableCommandRunner { node = coordinator.rootNode.descendantNodeRepresentingObject(representedObject as AnyObject) } - if let node = node, let indexPath = dataSource.indexPath(for: node), let unreadCountProvider = node.representedObject as? UnreadCountProvider { - if let cell = tableView.cellForRow(at: indexPath) as? MasterFeedTableViewCell { - cell.unreadCount = unreadCountProvider.unreadCount - } + if let node = node, dataSource.indexPath(for: node) != nil { + reloadNode(node) } }