diff --git a/src/sql/workbench/contrib/editorReplacement/test/browser/editorReplacerContribution.test.ts b/src/sql/workbench/contrib/editorReplacement/test/browser/editorReplacerContribution.test.ts
index 031347f900..e08c920625 100644
--- a/src/sql/workbench/contrib/editorReplacement/test/browser/editorReplacerContribution.test.ts
+++ b/src/sql/workbench/contrib/editorReplacement/test/browser/editorReplacerContribution.test.ts
@@ -22,7 +22,7 @@ import { Registry } from 'vs/platform/registry/common/platform';
 import { ILanguageAssociationRegistry, Extensions as LanguageAssociationExtensions } from 'sql/workbench/services/languageAssociation/common/languageAssociation';
 import { QueryEditorLanguageAssociation } from 'sql/workbench/contrib/query/browser/queryInputFactory';
 import { workbenchInstantiationService } from 'sql/workbench/test/workbenchTestServices';
-import { NotebookEditorInputAssociation } from 'sql/workbench/contrib/notebook/browser/models/nodebookInputFactory';
+import { NotebookEditorInputAssociation } from 'sql/workbench/contrib/notebook/browser/models/notebookInputFactory';
 import { NotebookInput } from 'sql/workbench/contrib/notebook/browser/models/notebookInput';
 import { UntitledTextEditorInput } from 'vs/workbench/services/untitled/common/untitledTextEditorInput';
 import { UntitledQueryEditorInput } from 'sql/workbench/common/editor/query/untitledQueryEditorInput';
diff --git a/src/sql/workbench/contrib/notebook/browser/models/nodebookInputFactory.ts b/src/sql/workbench/contrib/notebook/browser/models/notebookInputFactory.ts
similarity index 99%
rename from src/sql/workbench/contrib/notebook/browser/models/nodebookInputFactory.ts
rename to src/sql/workbench/contrib/notebook/browser/models/notebookInputFactory.ts
index 94ab5e2017..7f636341d9 100644
--- a/src/sql/workbench/contrib/notebook/browser/models/nodebookInputFactory.ts
+++ b/src/sql/workbench/contrib/notebook/browser/models/notebookInputFactory.ts
@@ -25,7 +25,7 @@ export class NotebookEditorInputAssociation implements ILanguageAssociation {
 
 	constructor(@IInstantiationService private readonly instantiationService: IInstantiationService, @IConfigurationService private readonly configurationService: IConfigurationService) { }
 
-	convertInput(activeEditor: IEditorInput): NotebookInput | DiffNotebookInput {
+	convertInput(activeEditor: IEditorInput): NotebookInput | DiffNotebookInput | undefined {
 		if (activeEditor instanceof FileEditorInput) {
 			return this.instantiationService.createInstance(FileNotebookInput, activeEditor.getName(), activeEditor.resource, activeEditor);
 		} else if (activeEditor instanceof UntitledTextEditorInput) {
diff --git a/src/sql/workbench/contrib/notebook/browser/notebook.contribution.ts b/src/sql/workbench/contrib/notebook/browser/notebook.contribution.ts
index b268bb4383..8edb280923 100644
--- a/src/sql/workbench/contrib/notebook/browser/notebook.contribution.ts
+++ b/src/sql/workbench/contrib/notebook/browser/notebook.contribution.ts
@@ -12,7 +12,7 @@ import { IEditorInputFactoryRegistry, Extensions as EditorInputFactoryExtensions
 import { ILanguageAssociationRegistry, Extensions as LanguageAssociationExtensions } from 'sql/workbench/services/languageAssociation/common/languageAssociation';
 import { UntitledNotebookInput } from 'sql/workbench/contrib/notebook/browser/models/untitledNotebookInput';
 import { FileNotebookInput } from 'sql/workbench/contrib/notebook/browser/models/fileNotebookInput';
-import { FileNoteBookEditorInputFactory, UntitledNoteBookEditorInputFactory, NotebookEditorInputAssociation } from 'sql/workbench/contrib/notebook/browser/models/nodebookInputFactory';
+import { FileNoteBookEditorInputFactory, UntitledNoteBookEditorInputFactory, NotebookEditorInputAssociation } from 'sql/workbench/contrib/notebook/browser/models/notebookInputFactory';
 import { IWorkbenchActionRegistry, Extensions as WorkbenchActionsExtensions } from 'vs/workbench/common/actions';
 import { SyncActionDescriptor, registerAction2, MenuRegistry, MenuId, Action2 } from 'vs/platform/actions/common/actions';
 
diff --git a/src/sql/workbench/test/browser/parts/editor/editorStatusModeSelect.test.ts b/src/sql/workbench/test/browser/parts/editor/editorStatusModeSelect.test.ts
index 3024ad7e06..58f52bf606 100644
--- a/src/sql/workbench/test/browser/parts/editor/editorStatusModeSelect.test.ts
+++ b/src/sql/workbench/test/browser/parts/editor/editorStatusModeSelect.test.ts
@@ -8,7 +8,7 @@ import * as sinon from 'sinon';
 import { setMode } from 'sql/workbench/browser/parts/editor/editorStatusModeSelect';
 import { IDisposable, dispose } from 'vs/base/common/lifecycle';
 import { QueryEditorLanguageAssociation } from 'sql/workbench/contrib/query/browser/queryInputFactory';
-import { NotebookEditorInputAssociation } from 'sql/workbench/contrib/notebook/browser/models/nodebookInputFactory';
+import { NotebookEditorInputAssociation } from 'sql/workbench/contrib/notebook/browser/models/notebookInputFactory';
 import { workbenchInstantiationService } from 'sql/workbench/test/workbenchTestServices';
 import { INotebookService } from 'sql/workbench/services/notebook/browser/notebookService';
 import { IEditorService } from 'vs/workbench/services/editor/common/editorService';