diff --git a/overlays/ki.diff b/overlays/ki.diff
index abd88bc..6247969 100644
--- a/overlays/ki.diff
+++ b/overlays/ki.diff
@@ -1,5 +1,5 @@
 diff --git a/ki/__init__.py b/ki/__init__.py
-index 3f29c1a..1d5c101 100644
+index 3f29c1a..17c820c 100644
 --- a/ki/__init__.py
 +++ b/ki/__init__.py
 @@ -405,7 +405,7 @@ def parse_note(parser: Lark, transformer: NoteTransformer, delta: Delta) -> Deck
@@ -11,6 +11,24 @@ index 3f29c1a..1d5c101 100644
      )
  
  
+@@ -466,7 +466,7 @@ def update_note(
+ 
+     nid = note.id
+     note.tags = decknote.tags
+-    note.flush()
++    note.col.update_note(note)
+ 
+     # Set the deck of the given note, as well as all its cards, and create a
+     # deck with this name if it doesn't already exist. See the
+@@ -496,7 +496,7 @@ def update_note(
+     warnings = map(lambda k: NoteFieldValidationWarning(nid, k, new_notetype), missing)
+     fields = [(key, field) for key, field in decknote.fields.items() if key in note]
+     stardo(update_field(decknote, note), fields)
+-    note.flush()
++    note.col.update_note(note)
+ 
+     # Remove if unhealthy.
+     fwarns: List[Warning] = check_fields_health(note)
 @@ -1510,14 +1510,15 @@ def _pull2(kirepo: KiRepo, col: Collection) -> Collection: