diff --git a/src/vs/workbench/contrib/notebook/common/notebookEditorInput.ts b/src/vs/workbench/contrib/notebook/common/notebookEditorInput.ts
index 89c356f12e9..2e92193c670 100644
--- a/src/vs/workbench/contrib/notebook/common/notebookEditorInput.ts
+++ b/src/vs/workbench/contrib/notebook/common/notebookEditorInput.ts
@@ -86,7 +86,7 @@ export class NotebookEditorInput extends AbstractResourceEditorInput {
 		});
 
 		this._register(extensionService.onWillStop(e => {
-			if (!this.isDirty()) {
+			if (!e.auto && !this.isDirty()) {
 				return;
 			}