diff --git a/common/python/ax/cluster_management/app/cluster_upgrader.py b/common/python/ax/cluster_management/app/cluster_upgrader.py
index 7a3f7fdb2542..e843decdf02b 100644
--- a/common/python/ax/cluster_management/app/cluster_upgrader.py
+++ b/common/python/ax/cluster_management/app/cluster_upgrader.py
@@ -15,6 +15,7 @@
 import yaml
 from pprint import pformat
 
+from ax.cloud import Cloud
 from ax.cloud.aws import AWS_DEFAULT_PROFILE
 from ax.meta import AXCustomerId
 from ax.platform.ax_cluster_info import AXClusterInfo
@@ -187,7 +188,8 @@ def _upgrade_kube(self):
             "OLD_KUBE_VERSION": self._current_software_info.kube_version,
             "NEW_KUBE_VERSION": self._cfg.target_software_info.kube_version,
             "NEW_CLUSTER_INSTALL_VERSION": self._cfg.target_software_info.kube_installer_version,
-            "ARGO_AWS_REGION": self._cluster_config.get_region()
+            "ARGO_AWS_REGION": self._cluster_config.get_region(),
+            "AX_TARGET_CLOUD": Cloud().target_cloud()
         }
         
         if self._cfg.cloud_profile:
@@ -199,7 +201,6 @@ def _upgrade_kube(self):
         env.update(os.environ)
         subprocess.check_call(["upgrade-kubernetes"], env=env)
 
-
     def _start_platform(self):
         """
         This step brings up Argo platform services