diff --git a/src/sql/workbench/contrib/notebook/browser/models/notebookInput.ts b/src/sql/workbench/contrib/notebook/browser/models/notebookInput.ts
index 85457f167a..1719f5b1e6 100644
--- a/src/sql/workbench/contrib/notebook/browser/models/notebookInput.ts
+++ b/src/sql/workbench/contrib/notebook/browser/models/notebookInput.ts
@@ -41,6 +41,7 @@ import { NotebookLanguage } from 'sql/workbench/common/constants';
 import { ITextResourcePropertiesService } from 'vs/editor/common/services/textResourceConfiguration';
 import { IEditorResolverService } from 'vs/workbench/services/editor/common/editorResolverService';
 import { isEqual } from 'vs/base/common/resources';
+import { NotebookEditor } from 'sql/workbench/contrib/notebook/browser/notebookEditor';
 
 export type ModeViewSaveHandler = (handle: number) => Thenable<boolean>;
 const languageAssociationRegistry = Registry.as<ILanguageAssociationRegistry>(LanguageAssociationExtensions.LanguageAssociations);
@@ -261,6 +262,10 @@ export abstract class NotebookInput extends EditorInput implements INotebookInpu
 		}
 	}
 
+	public override get editorId(): string {
+		return NotebookEditor.ID;
+	}
+
 	public get languageMode(): string {
 		return this._textInput.getLanguageId();
 	}