diff --git a/src/vs/workbench/api/browser/mainThreadNotebook.ts b/src/vs/workbench/api/browser/mainThreadNotebook.ts
index 0cb8f140776..55ea5fae139 100644
--- a/src/vs/workbench/api/browser/mainThreadNotebook.ts
+++ b/src/vs/workbench/api/browser/mainThreadNotebook.ts
@@ -102,11 +102,6 @@ class DocumentAndEditorState {
 		// const oldActiveEditor = before.activeEditor !== after.activeEditor ? before.activeEditor : undefined;
 		const newActiveEditor = before.activeEditor !== after.activeEditor ? after.activeEditor : undefined;
 
-		// return new DocumentAndEditorStateDelta(
-		// 	documentDelta.removed, documentDelta.added,
-		// 	editorDelta.removed, editorDelta.added,
-		// 	oldActiveEditor, newActiveEditor
-		// );
 		return {
 			addedEditors: addedAPIEditors,
 			removedEditors: removedAPIEditors,
@@ -523,11 +518,6 @@ export class MainThreadNotebookController implements IMainNotebookController {
 		document?.textModel.updateNotebookCellMetadata(handle, metadata);
 	}
 
-	// updateNotebookRenderers(resource: UriComponents, renderers: number[]): void {
-	// 	let document = this._mapping.get(URI.from(resource).toString());
-	// 	document?.textModel.updateRenderers(renderers);
-	// }
-
 	async executeNotebookCell(uri: URI, handle: number, useAttachedKernel: boolean, token: CancellationToken): Promise<void> {
 		return this._proxy.$executeNotebook(this._viewType, uri, handle, useAttachedKernel, token);
 	}
diff --git a/src/vs/workbench/contrib/notebook/browser/notebookServiceImpl.ts b/src/vs/workbench/contrib/notebook/browser/notebookServiceImpl.ts
index b0e43b6c244..e68645dc4c6 100644
--- a/src/vs/workbench/contrib/notebook/browser/notebookServiceImpl.ts
+++ b/src/vs/workbench/contrib/notebook/browser/notebookServiceImpl.ts
@@ -291,6 +291,7 @@ export class NotebookService extends Disposable implements INotebookService, ICu
 		}
 
 		const notebookModel = await provider.controller.createNotebook(viewType, uri, { metadata, languages, cells }, false, editorId);
+		await this.transformTextModelOutputs(notebookModel!);
 		if (!notebookModel) {
 			return undefined;
 		}
diff --git a/src/vs/workbench/contrib/notebook/common/model/notebookTextModel.ts b/src/vs/workbench/contrib/notebook/common/model/notebookTextModel.ts
index 7891148fa98..ef96c12335e 100644
--- a/src/vs/workbench/contrib/notebook/common/model/notebookTextModel.ts
+++ b/src/vs/workbench/contrib/notebook/common/model/notebookTextModel.ts
@@ -43,7 +43,8 @@ export class NotebookTextModelSnapshot implements ITextSnapshot {
 				source: cell.getValue(),
 				metadata: cell.metadata,
 				cellKind: cell.cellKind,
-				language: cell.language
+				language: cell.language,
+				outputs: cell.outputs
 			};
 
 			const rawStr = JSON.stringify(data);