Fix LocationDataSource crash in MockRelease build

1 files changed, 13 insertions, 8 deletions

diff --git a/ios/MullvadVPN/View controllers/SelectLocation/LocationDataSource.swift b/ios/MullvadVPN/View controllers/SelectLocation/LocationDataSource.swift
index 1a026e54e9..a8bd237226 100644
--- a/ios/MullvadVPN/View controllers/SelectLocation/LocationDataSource.swift
+++ b/ios/MullvadVPN/View controllers/SelectLocation/LocationDataSource.swift
@@ -26,7 +26,10 @@ final class LocationDataSource: UITableViewDiffableDataSource<SelectLocationSect
         customLists: LocationDataSourceProtocol
     ) {
         self.tableView = tableView
+
+        #if DEBUG
         self.dataSources.append(customLists)
+        #endif
         self.dataSources.append(allLocations)
 
         let locationCellFactory = LocationCellFactory(
@@ -76,14 +79,16 @@ final class LocationDataSource: UITableViewDiffableDataSource<SelectLocationSect
                 .map { LocationCellViewModel(group: group, location: $0) }
         }
 
-        updateDataSnapshot(with: list, reloadExisting: !searchString.isEmpty)
-
-        if searchString.isEmpty {
-            setSelectedRelayLocation(selectedRelayLocation, animated: false, completion: {
-                self.scrollToSelectedRelay()
-            })
-        } else {
-            scrollToTop(animated: false)
+        updateDataSnapshot(with: list, reloadExisting: !searchString.isEmpty) {
+            DispatchQueue.main.async {
+                if searchString.isEmpty {
+                    self.setSelectedRelayLocation(self.selectedRelayLocation, animated: false, completion: {
+                        self.scrollToSelectedRelay()
+                    })
+                } else {
+                    self.scrollToTop(animated: false)
+                }
+            }
         }
     }