The plugin does not use the correct endpoints when az cli is connected to AzureUSGovernment

[crstop]

After using az cloud set --name AzureUSGovernment then login with az login and verifying the correct environment with az account show --query environmentName using kubectl aks config import --subscription --resource-group --cluster-name I get and error

Error: getting VMSS VMs via Azure API: getting cluster: GET https://management.azure.com/subscriptions/<sub-id>/resourceGroups/<rg-name>/providers/Microsoft.ContainerService/managedClusters/<cluster-name>

I would expect the plugin to try to use the management endpoint from az cloud show --name AzureUSGovernment