diff --git a/src/vs/workbench/contrib/notebook/browser/viewModel/notebookViewModelImpl.ts b/src/vs/workbench/contrib/notebook/browser/viewModel/notebookViewModelImpl.ts
index 3e42b3fd7b2..385a37ad948 100644
--- a/src/vs/workbench/contrib/notebook/browser/viewModel/notebookViewModelImpl.ts
+++ b/src/vs/workbench/contrib/notebook/browser/viewModel/notebookViewModelImpl.ts
@@ -716,6 +716,7 @@ export class NotebookViewModel extends Disposable implements EditorFoldingStateD
 			if (handle !== undefined) {
 				const cell = this.getCellByHandle(handle);
 				cell?.deltaCellDecorations([id], []);
+				this._decorationIdToCellMap.delete(id);
 			}
 		});