diff --git a/src/sql/workbench/browser/modelComponents/tree.component.ts b/src/sql/workbench/browser/modelComponents/tree.component.ts
index 1e442ea846..a1b6a074d5 100644
--- a/src/sql/workbench/browser/modelComponents/tree.component.ts
+++ b/src/sql/workbench/browser/modelComponents/tree.component.ts
@@ -28,6 +28,7 @@ import { IThemeService } from 'vs/platform/theme/common/themeService';
 import { IComponentDescriptor, IComponent, IModelStore } from 'sql/platform/dashboard/browser/interfaces';
 import { convertSizeToNumber } from 'sql/base/browser/dom';
 import { ILogService } from 'vs/platform/log/common/log';
+import { INotificationService } from 'vs/platform/notification/common/notification';
 
 class Root implements ITreeComponentItem {
 	label = {
@@ -59,7 +60,8 @@ export default class TreeComponent extends ComponentBase<azdata.TreeProperties>
 		@Inject(IThemeService) private themeService: IThemeService,
 		@Inject(IInstantiationService) private _instantiationService: IInstantiationService,
 		@Inject(forwardRef(() => ElementRef)) el: ElementRef,
-		@Inject(ILogService) logService: ILogService
+		@Inject(ILogService) logService: ILogService,
+		@Inject(INotificationService) private _notificationService: INotificationService
 	) {
 		super(changeRef, el, logService);
 	}
@@ -76,7 +78,7 @@ export default class TreeComponent extends ComponentBase<azdata.TreeProperties>
 	}
 
 	public setDataProvider(handle: number, componentId: string, context: any): any {
-		this._dataProvider = new TreeViewDataProvider(handle, componentId, context);
+		this._dataProvider = new TreeViewDataProvider(handle, componentId, context, this._notificationService);
 		this.createTreeControl();
 	}
 
diff --git a/src/sql/workbench/browser/modelComponents/treeViewDataProvider.ts b/src/sql/workbench/browser/modelComponents/treeViewDataProvider.ts
index 5f63c16dd1..fc554cab9d 100644
--- a/src/sql/workbench/browser/modelComponents/treeViewDataProvider.ts
+++ b/src/sql/workbench/browser/modelComponents/treeViewDataProvider.ts
@@ -33,7 +33,7 @@ export class ResolvableTreeComponentItem extends ResolvableTreeItem implements I
 export class TreeViewDataProvider extends vsTreeView.TreeViewDataProvider implements IModelViewTreeViewDataProvider {
 	constructor(handle: number, treeViewId: string,
 		context: IExtHostContext,
-		notificationService?: INotificationService
+		notificationService: INotificationService
 	) {
 		super(`${handle}-${treeViewId}`, context.getProxy(SqlExtHostContext.ExtHostModelViewTreeViews), notificationService);
 	}