diff --git a/Tusker/Screens/Timeline/TimelineViewController.swift b/Tusker/Screens/Timeline/TimelineViewController.swift
index 3479abab..c773e226 100644
--- a/Tusker/Screens/Timeline/TimelineViewController.swift
+++ b/Tusker/Screens/Timeline/TimelineViewController.swift
@@ -420,7 +420,7 @@ class TimelineViewController: UIViewController, TimelineLikeCollectionViewContro
                 }
                 let hasStatusesToRestore = await loadStatusesToRestore(position: position)
                 if hasStatusesToRestore {
-                    applyItemsToRestore(position: position)
+                    await applyItemsToRestore(position: position)
                     loaded = true
                 }
             case .mastodon:
@@ -443,7 +443,7 @@ class TimelineViewController: UIViewController, TimelineLikeCollectionViewContro
             position.centerStatusID = centerStatusID
             let hasStatusesToRestore = await loadStatusesToRestore(position: position)
             if hasStatusesToRestore {
-                applyItemsToRestore(position: position)
+                await applyItemsToRestore(position: position)
             }
             mastodonController.persistentContainer.viewContext.delete(position)
         }
@@ -532,7 +532,7 @@ class TimelineViewController: UIViewController, TimelineLikeCollectionViewContro
     }
     
     @MainActor
-    private func applyItemsToRestore(position: TimelinePosition) {
+    private func applyItemsToRestore(position: TimelinePosition) async {
         var snapshot = NSDiffableDataSourceSnapshot<Section, Item>()
         snapshot.appendSections([.statuses])
         let statusIDs = position.statusIDs
@@ -545,14 +545,13 @@ class TimelineViewController: UIViewController, TimelineLikeCollectionViewContro
             "statusIDs": position.statusIDs
         ]
         SentrySDK.addBreadcrumb(crumb)
-        dataSource.apply(snapshot, animatingDifferences: false) {
-            if let centerStatusID,
-               let index = statusIDs.firstIndex(of: centerStatusID) {
-                self.scrollToItem(item: items[index])
-                stateRestorationLogger.info("TimelineViewController: restored statuses with center ID \(centerStatusID)")
-            } else {
-                stateRestorationLogger.fault("TimelineViewController: restored statuses, but couldn't find center ID")
-            }
+        await apply(snapshot, animatingDifferences: false)
+        if let centerStatusID,
+           let index = statusIDs.firstIndex(of: centerStatusID) {
+            self.scrollToItem(item: items[index])
+            stateRestorationLogger.info("TimelineViewController: restored statuses with center ID \(centerStatusID)")
+        } else {
+            stateRestorationLogger.fault("TimelineViewController: restored statuses, but couldn't find center ID")
         }
     }