avoid null issue

districtr · Nov 11, 2024 · e11426a · e11426a
1 parent 11ad326
commit e11426a
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/app/src/app/store/mapStore.ts b/app/src/app/store/mapStore.ts
@@ -382,11 +382,12 @@ export const useMapStore = create(
     ),
     {
       diff: (pastState: Partial<MapStore>, currentState: Partial<MapStore>) => {
-        currentState.zoneAssignments.keys().forEach((geoid) => {
+        if (!currentState.zoneAssignments || !pastState.zoneAssignments) return pastState;
+        for (const geoid of currentState.zoneAssignments.keys()) {
           if (!pastState.zoneAssignments.has(geoid)) {
             pastState.zoneAssignments.set(geoid, null);
           }
-        });
+        }
         return pastState as Partial<MapStore>;
       },
       equality: (pastState, currentState) => {