Remove restoration identifier when reloading

hotwired · Jul 4, 2024 · 014e4f6 · 014e4f6
1 parent 23d5ca9
commit 014e4f6
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/Source/Session/Session.swift b/Source/Session/Session.swift
@@ -61,6 +61,7 @@ public class Session: NSObject {
 
         if reload {
             initialized = false
+            removeRestorationIdentifier(for: visitable)
         }
 
         let visit = makeVisit(for: visitable, options: options ?? VisitOptions())
@@ -150,6 +151,10 @@ public class Session: NSObject {
         visitableRestorationIdentifiers.setObject(restorationIdentifier as NSString, forKey: visitable.visitableViewController)
     }
 
+    private func removeRestorationIdentifier(for visitable: Visitable) {
+        visitableRestorationIdentifiers.removeObject(forKey: visitable.visitableViewController)
+    }
+
     // MARK: - Navigation
 
     private func completeNavigationForCurrentVisit() {