fixes

dotJEM · Nov 29, 2023 · c08436e · c08436e
1 parent 4500b47
commit c08436e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/DotJEM.Json.Index2.Snapshots/IndexSnapshotHandler.cs b/src/DotJEM.Json.Index2.Snapshots/IndexSnapshotHandler.cs
@@ -90,7 +90,7 @@ public async Task<ISnapshot> RestoreSnapshotAsync(IJsonIndex index, ISnapshot sn
         segmentsSourceStream.Dispose();
         dir.Sync(new [] { segmentsFile.Name });
 
-        SegmentInfos.WriteSegmentsGen(dir, reader.Snapshot.Generation);
+        SegmentInfos.WriteSegmentsGen(dir, snapshot.Generation);
 
         //NOTE: (jmd 2020-09-30) Not quite sure what this does at this time, but the Lucene Replicator does it so better keep it for now.
         IndexCommit last = DirectoryReader.ListCommits(dir).Last();
@@ -100,6 +100,6 @@ public async Task<ISnapshot> RestoreSnapshotAsync(IJsonIndex index, ISnapshot sn
             commitFiles.Add(IndexFileNames.SEGMENTS_GEN);
         }
         index.WriterManager.Close();
-        return reader.Snapshot;
+        return snapshot;
     }
 }