1 files changed, 3 insertions, 3 deletions
diff --git a/Firestore/Source/Local/FSTLocalStore.mm b/Firestore/Source/Local/FSTLocalStore.mm
index 2ea3328..7508358 100644
--- a/Firestore/Source/Local/FSTLocalStore.mm
+++ b/Firestore/Source/Local/FSTLocalStore.mm
@@ -393,9 +393,9 @@ NS_ASSUME_NONNULL_BEGIN
 }
 
 - (nullable FSTMutationBatch *)nextMutationBatchAfterBatchID:(FSTBatchID)batchID {
-  FSTWriteGroup *group = [self.persistence startGroupWithAction:@"nextMutationBatchAfterBatchID"];
-  FSTMutationBatch *result = [self.mutationQueue nextMutationBatchAfterBatchID:batchID];
-  [self.persistence commitGroup:group];
+  FSTMutationBatch *result = self.persistence.run([&]() -> FSTMutationBatch * {
+    return [self.mutationQueue nextMutationBatchAfterBatchID:batchID];
+  });
   return result;
 }