From aa0bc1cab906ad69a04de0bac1845642cfbe8171 Mon Sep 17 00:00:00 2001
From: Jason Lubken <jlubken@users.noreply.github.com>
Date: Mon, 4 May 2020 16:37:12 -0400
Subject: [PATCH] Fix mongo batch record.end update

---
 src/dsdk/mongo.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/dsdk/mongo.py b/src/dsdk/mongo.py
index 05653ca..bda9555 100644
--- a/src/dsdk/mongo.py
+++ b/src/dsdk/mongo.py
@@ -106,9 +106,9 @@ def open_batch(
 
             yield batch
 
-            key, doc = batch.as_update_doc()
-            with self.open_mongo() as database:
-                update_one(database.batches, key, doc)
+        key, doc = batch.as_update_doc()
+        with self.open_mongo() as database:
+            update_one(database.batches, key, doc)
 
     def store_evidence(self, batch: Batch, *args, **kwargs) -> None:
         """Store Evidence."""