diff --git a/src/k8s-extension/azext_k8s_extension/partner_extensions/AzureMLKubernetes.py b/src/k8s-extension/azext_k8s_extension/partner_extensions/AzureMLKubernetes.py
index f985bec7541..88d2595796b 100644
--- a/src/k8s-extension/azext_k8s_extension/partner_extensions/AzureMLKubernetes.py
+++ b/src/k8s-extension/azext_k8s_extension/partner_extensions/AzureMLKubernetes.py
@@ -83,7 +83,7 @@ def __init__(self):
 
         self.inferenceRouterServiceType = 'inferenceRouterServiceType'
         self.internalLoadBalancerProvider = 'internalLoadBalancerProvider'
-        self.inferenceLoadBalancerHA = 'inferenceLoadBalancerHA'
+        self.inferenceRouterHA = 'inferenceRouterHA'
 
         # constants for existing AKS to AMLARC migration
         self.IS_AKS_MIGRATION = 'isAKSMigration'
@@ -322,14 +322,13 @@ def Update(self, cmd, resource_group_name, cluster_name, auto_upgrade_minor_vers
 
     def __normalize_config(self, configuration_settings, configuration_protected_settings):
         # inference
-        isTestCluster = _get_value_from_config_protected_config(
-            self.inferenceLoadBalancerHA, configuration_settings, configuration_protected_settings)
-        if isTestCluster is not None:
-            isTestCluster = str(isTestCluster).lower() == 'false'
-            if isTestCluster:
-                configuration_settings['clusterPurpose'] = 'DevTest'
-            else:
-                configuration_settings['clusterPurpose'] = 'FastProd'
+        inferenceRouterHA = _get_value_from_config_protected_config(
+            self.inferenceRouterHA, configuration_settings, configuration_protected_settings)
+        isTestCluster = True if inferenceRouterHA is not None and str(inferenceRouterHA).lower() == 'false' else False
+        if isTestCluster:
+            configuration_settings['clusterPurpose'] = 'DevTest'
+        else:
+            configuration_settings['clusterPurpose'] = 'FastProd'
 
         inferenceRouterServiceType = _get_value_from_config_protected_config(
             self.inferenceRouterServiceType, configuration_settings, configuration_protected_settings)
@@ -386,9 +385,9 @@ def __validate_config(self, configuration_settings, configuration_protected_sett
         configuration_protected_settings.pop(self.ENABLE_INFERENCE, None)
 
     def __validate_scoring_fe_settings(self, configuration_settings, configuration_protected_settings, release_namespace):
-        isTestCluster = _get_value_from_config_protected_config(
-            self.inferenceLoadBalancerHA, configuration_settings, configuration_protected_settings)
-        isTestCluster = str(isTestCluster).lower() == 'false'
+        inferenceRouterHA = _get_value_from_config_protected_config(
+            self.inferenceRouterHA, configuration_settings, configuration_protected_settings)
+        isTestCluster = True if inferenceRouterHA is not None and str(inferenceRouterHA).lower() == 'false' else False
         if isTestCluster:
             configuration_settings['clusterPurpose'] = 'DevTest'
         else: