diff --git a/iOS/MasterFeed/MasterFeedViewController.swift b/iOS/MasterFeed/MasterFeedViewController.swift index 7d2cb54ae..377ba873d 100644 --- a/iOS/MasterFeed/MasterFeedViewController.swift +++ b/iOS/MasterFeed/MasterFeedViewController.swift @@ -410,25 +410,33 @@ class MasterFeedViewController: UITableViewController, UndoableCommandRunner { @objc func expandSelectedRows(_ sender: Any?) { if let indexPath = coordinator.currentFeedIndexPath { coordinator.expandFolder(indexPath) - self.applyChanges(animate: true) + self.applyChanges(animate: true) { + self.reloadAllVisibleCells() + } } } @objc func collapseSelectedRows(_ sender: Any?) { if let indexPath = coordinator.currentFeedIndexPath { coordinator.collapseFolder(indexPath) - self.applyChanges(animate: true) + self.applyChanges(animate: true) { + self.reloadAllVisibleCells() + } } } @objc func expandAll(_ sender: Any?) { coordinator.expandAllSectionsAndFolders() - self.applyChanges(animate: true) + self.applyChanges(animate: true) { + self.reloadAllVisibleCells() + } } @objc func collapseAllExceptForGroupItems(_ sender: Any?) { coordinator.collapseAllFolders() - self.applyChanges(animate: true) + self.applyChanges(animate: true) { + self.reloadAllVisibleCells() + } } // MARK: API