diff --git a/src/vs/workbench/contrib/notebook/browser/viewModel/cellEdit.ts b/src/vs/workbench/contrib/notebook/browser/viewModel/cellEdit.ts
index cc3aa9cbca9..6c73c4348ff 100644
--- a/src/vs/workbench/contrib/notebook/browser/viewModel/cellEdit.ts
+++ b/src/vs/workbench/contrib/notebook/browser/viewModel/cellEdit.ts
@@ -22,7 +22,7 @@ export interface IViewCellEditingDelegate extends ITextCellEditingDelegate {
 export class JoinCellEdit implements IResourceUndoRedoElement {
 	type: UndoRedoElementType.Resource = UndoRedoElementType.Resource;
 	label: string = 'Join Cell';
-	code: string = 'undoredo.notebooks.joinCell';
+	code: string = 'undoredo.textBufferEdit';
 	private _deletedRawCell: NotebookCellTextModel;
 	constructor(
 		public resource: URI,
diff --git a/src/vs/workbench/contrib/notebook/common/model/cellEdit.ts b/src/vs/workbench/contrib/notebook/common/model/cellEdit.ts
index 45eccd656c3..0625af50b5d 100644
--- a/src/vs/workbench/contrib/notebook/common/model/cellEdit.ts
+++ b/src/vs/workbench/contrib/notebook/common/model/cellEdit.ts
@@ -24,7 +24,7 @@ export class MoveCellEdit implements IResourceUndoRedoElement {
 	get label() {
 		return this.length === 1 ? 'Move Cell' : 'Move Cells';
 	}
-	code: string = 'undoredo.notebooks.moveCell';
+	code: string = 'undoredo.textBufferEdit';
 
 	constructor(
 		public resource: URI,
@@ -67,7 +67,7 @@ export class SpliceCellsEdit implements IResourceUndoRedoElement {
 		// Default to Insert Cell
 		return 'Insert Cell';
 	}
-	code: string = 'undoredo.notebooks.insertCell';
+	code: string = 'undoredo.textBufferEdit';
 	constructor(
 		public resource: URI,
 		private diffs: [number, NotebookCellTextModel[], NotebookCellTextModel[]][],
diff --git a/src/vs/workbench/contrib/notebook/common/model/notebookTextModel.ts b/src/vs/workbench/contrib/notebook/common/model/notebookTextModel.ts
index e78ccadce83..3f57341e228 100644
--- a/src/vs/workbench/contrib/notebook/common/model/notebookTextModel.ts
+++ b/src/vs/workbench/contrib/notebook/common/model/notebookTextModel.ts
@@ -1026,7 +1026,7 @@ export class NotebookTextModel extends Disposable implements INotebookTextModel
 					return that.uri;
 				}
 				readonly label = 'Update Cell Language';
-				readonly code = 'undoredo.notebooks.updateCellLanguage';
+				readonly code = 'undoredo.textBufferEdit';
 				undo() {
 					that._changeCellLanguage(cell, oldLanguage, false, beginSelectionState, undoRedoGroup);
 				}