Make sure to end undo group (#3)

ChimeHQ · Sep 18, 2023 · c92768d · c92768d
1 parent 10810f3
commit c92768d
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/Sources/TextFormationPlugin/TextFormationPlugin.swift b/Sources/TextFormationPlugin/TextFormationPlugin.swift
@@ -66,6 +66,9 @@ public struct TextFormationPlugin: STPlugin {
 			let mutation = TextMutation(string: string, range: range, limit: limit)
 
 			textView.undoManager?.beginUndoGrouping()
+			defer {
+				textView.undoManager?.endUndoGrouping()
+			}
 
 			for filter in filters {
 				switch filter.processMutation(mutation, in: adapter, with: whitespaceProviders) {
@@ -78,8 +81,6 @@ public struct TextFormationPlugin: STPlugin {
 				}
 			}
 
-			textView.undoManager?.endUndoGrouping()
-
             return true
 		}
 	}