diff --git a/src/main.ts b/src/main.ts
index cd355f0e6..6b0bbfd59 100644
--- a/src/main.ts
+++ b/src/main.ts
@@ -34,6 +34,7 @@ async function main() {
             throw new Error("Not all values are present in the creds object. Ensure clientId, clientSecret, tenantId and subscriptionId are supplied.");
         }
         // Attempting Az cli login
+        await executeAzCliCommand('set --name AzureUSGovernment', true);
         await executeAzCliCommand(`login --service-principal -u "${servicePrincipalId}" -p "${servicePrincipalKey}" --tenant "${tenantId}"`, true);
         await executeAzCliCommand(`account set --subscription "${subscriptionId}"`, true);
         isAzCLISuccess = true;