From 59e71a02cc7c65ef794afd07c06bd0090b8f09a1 Mon Sep 17 00:00:00 2001
From: Patrick Trillsam <ptrillsam@seven.com.au>
Date: Tue, 26 Aug 2025 07:47:21 +1000
Subject: [PATCH] Fix reloading after sections updates

---
 MagazineLayout/Public/MagazineLayout.swift | 32 +++++++++-------------
 1 file changed, 13 insertions(+), 19 deletions(-)

diff --git a/MagazineLayout/Public/MagazineLayout.swift b/MagazineLayout/Public/MagazineLayout.swift
index 4842215..2e30076 100755
--- a/MagazineLayout/Public/MagazineLayout.swift
+++ b/MagazineLayout/Public/MagazineLayout.swift
@@ -102,6 +102,19 @@ public final class MagazineLayout: UICollectionViewLayout {
       prepareActions.contains(.recreateSectionModels)
     {
       hasPinnedHeaderOrFooter = false
+      lastSizedElementMinY = nil
+      lastSizedElementPreferredHeight = nil
+    }
+
+    // Recreate section models from scratch if necessary
+    if prepareActions.contains(.recreateSectionModels) {
+      var sections = [SectionModel]()
+      for sectionIndex in 0..<currentCollectionView.numberOfSections {
+        let sectionModel = sectionModelForSection(atIndex: sectionIndex)
+        sections.append(sectionModel)
+      }
+
+      modelState.setSections(sections)
     }
 
     // Update layout metrics if necessary
@@ -136,25 +149,6 @@ public final class MagazineLayout: UICollectionViewLayout {
       }
     }
 
-    // Recreate section models from scratch if necessary
-    if prepareActions.contains(.recreateSectionModels) {
-      var sections = [SectionModel]()
-      for sectionIndex in 0..<currentCollectionView.numberOfSections {
-        let sectionModel = sectionModelForSection(atIndex: sectionIndex)
-        sections.append(sectionModel)
-      }
-
-      modelState.setSections(sections)
-    }
-
-    if
-      prepareActions.contains(.recreateSectionModels) ||
-      prepareActions.contains(.updateLayoutMetrics)
-    {
-      lastSizedElementMinY = nil
-      lastSizedElementPreferredHeight = nil
-    }
-
     prepareActions = []
   }