diff --git a/src/vs/editor/browser/widget/diffEditorWidget.ts b/src/vs/editor/browser/widget/diffEditorWidget.ts
index c98831decf12a..bc2d3e3bd5f88 100644
--- a/src/vs/editor/browser/widget/diffEditorWidget.ts
+++ b/src/vs/editor/browser/widget/diffEditorWidget.ts
@@ -623,7 +623,7 @@ export class DiffEditorWidget extends Disposable implements editorBrowser.IDiffE
 		}));
 
 		// Revert change when an arrow is clicked.
-		this._register(editor.onMouseUp(event => {
+		this._register(editor.onMouseDown(event => {
 			if (!event.event.rightButton && event.target.position && event.target.element?.className.includes('arrow-revert-change')) {
 				const lineNumber = event.target.position.lineNumber;
 				const viewZone = event.target as editorBrowser.IMouseTargetViewZone | undefined;
diff --git a/src/vs/editor/browser/widget/inlineDiffMargin.ts b/src/vs/editor/browser/widget/inlineDiffMargin.ts
index 700949147e855..69dfd17560cb5 100644
--- a/src/vs/editor/browser/widget/inlineDiffMargin.ts
+++ b/src/vs/editor/browser/widget/inlineDiffMargin.ts
@@ -190,7 +190,7 @@ export class InlineDiffMargin extends Disposable {
 			}
 		}));
 
-		this._register(editor.onMouseUp((e: IEditorMouseEvent) => {
+		this._register(editor.onMouseDown((e: IEditorMouseEvent) => {
 			if (!e.event.rightButton) {
 				return;
 			}