diff --git a/extensions/azurecore/src/account-provider/auths/azureDeviceCode.ts b/extensions/azurecore/src/account-provider/auths/azureDeviceCode.ts index 01da492b88..df9bf180fd 100644 --- a/extensions/azurecore/src/account-provider/auths/azureDeviceCode.ts +++ b/extensions/azurecore/src/account-provider/auths/azureDeviceCode.ts @@ -51,7 +51,7 @@ export class AzureDeviceCode extends AzureAuth { context: vscode.ExtensionContext, uriEventEmitter: vscode.EventEmitter, ) { - super(metadata, tokenCache, context, uriEventEmitter, AzureAuthType.AuthCodeGrant, AzureDeviceCode.USER_FRIENDLY_NAME); + super(metadata, tokenCache, context, uriEventEmitter, AzureAuthType.DeviceCode, AzureDeviceCode.USER_FRIENDLY_NAME); this.pageTitle = localize('addAccount', "Add {0} account", this.metadata.displayName); } diff --git a/extensions/azurecore/src/account-provider/azureAccountProvider.ts b/extensions/azurecore/src/account-provider/azureAccountProvider.ts index 2915c24ed9..09d3d80e78 100644 --- a/extensions/azurecore/src/account-provider/azureAccountProvider.ts +++ b/extensions/azurecore/src/account-provider/azureAccountProvider.ts @@ -113,7 +113,7 @@ export class AzureAccountProvider implements azdata.AccountProvider, vscode.Disp private async _getAccountSecurityToken(account: AzureAccount, tenantId: string, resource: azdata.AzureResource): Promise { await this.initCompletePromise; - const azureAuth = this.getAuthMethod(undefined); + const azureAuth = this.getAuthMethod(account); Logger.pii(`Getting account security token for ${JSON.stringify(account.key)} (tenant ${tenantId}). Auth Method = ${azureAuth.userFriendlyName}`, [], []); return azureAuth?.getAccountSecurityToken(account, tenantId, resource); }