diff --git a/src/sql/workbench/services/notebook/browser/models/notebookModel.ts b/src/sql/workbench/services/notebook/browser/models/notebookModel.ts
index f8cd0f4ebc..cc48777538 100644
--- a/src/sql/workbench/services/notebook/browser/models/notebookModel.ts
+++ b/src/sql/workbench/services/notebook/browser/models/notebookModel.ts
@@ -712,6 +712,7 @@ export class NotebookModel extends Disposable implements INotebookModel {
 			if (this.isValidConnection(profile)) {
 				this._activeConnection = profile;
 				this._savedConnectionName = profile.connectionName;
+				kernelAlias = this._currentKernelAlias;
 			}
 
 			clientSession.onKernelChanging(async (e) => {