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 48a6b0dadb..c4a84720d5 100644
--- a/src/sql/workbench/contrib/editorReplacement/test/browser/editorReplacerContribution.test.ts
+++ b/src/sql/workbench/contrib/editorReplacement/test/browser/editorReplacerContribution.test.ts
@@ -13,7 +13,7 @@ import { URI } from 'vs/base/common/uri';
 import { IOpenEditorOverrideHandler, IOpenEditorOverride, IEditorService } from 'vs/workbench/services/editor/common/editorService';
 import { IDisposable, toDisposable, dispose } from 'vs/base/common/lifecycle';
 import { isUndefinedOrNull } from 'vs/base/common/types';
-import { IEditorInput, EditorInput, IUntitledTextResourceInput } from 'vs/workbench/common/editor';
+import { IEditorInput, EditorInput, IUntitledTextResourceEditorInput } from 'vs/workbench/common/editor';
 import { ITextEditorOptions, IEditorOptions } from 'vs/platform/editor/common/editor';
 import { IEditorGroup } from 'vs/workbench/services/editor/common/editorGroupsService';
 import { FileEditorInput } from 'vs/workbench/contrib/files/common/editors/fileEditorInput';
@@ -205,7 +205,7 @@ class MockEditorService extends TestEditorService {
 		return Promise.resolve(_editor);
 	}
 
-	createInput(_input: IUntitledTextResourceInput): EditorInput {
+	createEditorInput(_input: IUntitledTextResourceEditorInput): EditorInput {
 		const accessor = this.instantiationService.createInstance(ServiceAccessor);
 		const service = accessor.untitledTextEditorService;
 		return this.instantiationService.createInstance(UntitledTextEditorInput, service.create());
diff --git a/src/sql/workbench/services/queryEditor/test/common/testQueryEditorService.ts b/src/sql/workbench/services/queryEditor/test/common/testQueryEditorService.ts
index 456b87e7a6..98ac4c64b8 100644
--- a/src/sql/workbench/services/queryEditor/test/common/testQueryEditorService.ts
+++ b/src/sql/workbench/services/queryEditor/test/common/testQueryEditorService.ts
@@ -21,7 +21,7 @@ export class TestQueryEditorService implements IQueryEditorService {
 	}
 
 	newSqlEditor(sqlContent?: string, connectionProviderName?: string, isDirty?: boolean, objectName?: string): Promise<IConnectableInput> {
-		const base = this.editorService.createInput({ forceUntitled: true }) as UntitledTextEditorInput;
+		const base = this.editorService.createEditorInput({ forceUntitled: true }) as UntitledTextEditorInput;
 		return Promise.resolve(this.instantiationService.createInstance(UntitledQueryEditorInput, '', base, new QueryResultsInput(base.resource.toString(true))));
 	}