From 647d1f1167f8861641efb15a79a0be4cb665f985 Mon Sep 17 00:00:00 2001
From: Leila Lali <llali@microsoft.com>
Date: Mon, 9 Aug 2021 14:26:40 -0700
Subject: [PATCH] Fixed a bug with loading model list (#16640)

---
 .../src/views/models/manageModels/currentModelsComponent.ts    | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/extensions/machine-learning/src/views/models/manageModels/currentModelsComponent.ts b/extensions/machine-learning/src/views/models/manageModels/currentModelsComponent.ts
index f6cfae0f61..2177dbc257 100644
--- a/extensions/machine-learning/src/views/models/manageModels/currentModelsComponent.ts
+++ b/extensions/machine-learning/src/views/models/manageModels/currentModelsComponent.ts
@@ -148,6 +148,8 @@ export class CurrentModelsComponent extends ModelViewBase implements IPageView {
 
 				if (this._dataTable.isEmpty) {
 					formBuilder.addFormItem({ title: '', component: this._labelContainer });
+				} else {
+					formBuilder.removeFormItem({ title: '', component: this._labelContainer });
 				}
 				if (this._tableDataCountComponent) {
 					this._tableDataCountComponent.value = constants.getDataCount(this._dataTable.modelCounts);
@@ -200,7 +202,6 @@ export class CurrentModelsComponent extends ModelViewBase implements IPageView {
 
 	private refreshComponents(): void {
 		if (this._formBuilder) {
-			this.removeComponents(this._formBuilder);
 			this.addComponents(this._formBuilder);
 		}
 	}