redo adds back to same graph

BrickSchema · Aug 15, 2023 · 4172ba1 · 4172ba1
1 parent ecac3dd
commit 4172ba1
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/brickschema/persistent.py b/brickschema/persistent.py
@@ -169,10 +169,11 @@ def redo(self):
                 "SELECT * from changesets WHERE id = ?", (changeset_id,)
             ):
                 triple = pickle.loads(row["triple"])
+                graph = self.get_context(redo_record["graph"])
                 if row["is_insertion"]:
-                    self.remove((triple[0], triple[1], triple[2]))
+                    graph.remove((triple[0], triple[1], triple[2]))
                 else:
-                    self.add((triple[0], triple[1], triple[2]))
+                    graph.add((triple[0], triple[1], triple[2]))
 
     def versions(self, graph=None):
         """