diff --git a/extensions/sql-migration/package.json b/extensions/sql-migration/package.json
index 4d5e98fb80..7d38d383d1 100644
--- a/extensions/sql-migration/package.json
+++ b/extensions/sql-migration/package.json
@@ -2,7 +2,7 @@
 	"name": "sql-migration",
 	"displayName": "%displayName%",
 	"description": "%description%",
-	"version": "0.1.10",
+	"version": "0.1.11",
 	"publisher": "Microsoft",
 	"preview": true,
 	"license": "https://raw.githubusercontent.com/Microsoft/azuredatastudio/main/LICENSE.txt",
diff --git a/extensions/sql-migration/src/dialog/sqlMigrationService/sqlMigrationServiceDetailsDialog.ts b/extensions/sql-migration/src/dialog/sqlMigrationService/sqlMigrationServiceDetailsDialog.ts
index 463dc65bd9..e729254122 100644
--- a/extensions/sql-migration/src/dialog/sqlMigrationService/sqlMigrationServiceDetailsDialog.ts
+++ b/extensions/sql-migration/src/dialog/sqlMigrationService/sqlMigrationServiceDetailsDialog.ts
@@ -216,10 +216,10 @@ export class SqlMigrationServiceDetailsDialog {
 	}
 
 	private async _updateTableCell(table: azdata.DeclarativeTableComponent, row: number, col: number, value: string, keyName: string): Promise<void> {
-		const dataValues = table.dataValues;
+		const dataValues = table.dataValues!;
 		dataValues![row][col].value = value;
-		table.dataValues = [];
-		table.dataValues = dataValues;
+		await table.setDataValues([]);
+		await table.setDataValues(dataValues);
 		await vscode.window.showInformationMessage(constants.AUTH_KEY_REFRESHED(keyName));
 	}
 
diff --git a/extensions/sql-migration/src/wizard/databaseBackupPage.ts b/extensions/sql-migration/src/wizard/databaseBackupPage.ts
index b6d8d7702c..aca9cb738e 100644
--- a/extensions/sql-migration/src/wizard/databaseBackupPage.ts
+++ b/extensions/sql-migration/src/wizard/databaseBackupPage.ts
@@ -821,7 +821,7 @@ export class DatabaseBackupPage extends MigrationWizardPage {
 				this.migrationStateModel._targetDatabaseNames = [];
 				this.migrationStateModel._databaseBackup.blobs = [];
 				this.migrationStateModel._migrationDbs.forEach((db, index) => {
-					this.migrationStateModel._targetDatabaseNames.push('');
+					this.migrationStateModel._targetDatabaseNames.push(db);
 					this.migrationStateModel._databaseBackup.blobs.push(<Blob>{});
 					const targetDatabaseInput = this._view.modelBuilder.inputBox().withProps({
 						required: true,
@@ -973,7 +973,7 @@ export class DatabaseBackupPage extends MigrationWizardPage {
 					});
 					data.push(targetRow);
 				});
-				this._networkShareTargetDatabaseNamesTable.dataValues = data;
+				await this._networkShareTargetDatabaseNamesTable.setDataValues(data);
 
 				data = [];
 				this.migrationStateModel._migrationDbs.forEach((db, index) => {
diff --git a/extensions/sql-migration/src/wizard/integrationRuntimePage.ts b/extensions/sql-migration/src/wizard/integrationRuntimePage.ts
index 5318e86f1c..c1bea0f269 100644
--- a/extensions/sql-migration/src/wizard/integrationRuntimePage.ts
+++ b/extensions/sql-migration/src/wizard/integrationRuntimePage.ts
@@ -504,7 +504,7 @@ export class IntergrationRuntimePage extends MigrationWizardPage {
 					]
 				];
 
-				this._authKeyTable.dataValues = data;
+				await this._authKeyTable.setDataValues(data);
 			}
 		} catch (e) {
 			console.log(e);