diff --git a/magnum/drivers/heat/k8s_fedora_template_def.py b/magnum/drivers/heat/k8s_fedora_template_def.py
index bdf38e8207..d9263d6336 100644
--- a/magnum/drivers/heat/k8s_fedora_template_def.py
+++ b/magnum/drivers/heat/k8s_fedora_template_def.py
@@ -82,7 +82,7 @@ class K8sFedoraTemplateDefinition(k8s_template_def.K8sTemplateDefinition):
         if kube_tag:
             extra_params['kube_tag'] = kube_tag
 
-        container_infra_prefix = cluster_template.labels.get(
+        container_infra_prefix = cluster.labels.get(
             'container_infra_prefix')
         if container_infra_prefix:
             extra_params['container_infra_prefix'] = container_infra_prefix
diff --git a/magnum/tests/unit/drivers/test_template_definition.py b/magnum/tests/unit/drivers/test_template_definition.py
index 9523899961..2f308fba77 100644
--- a/magnum/tests/unit/drivers/test_template_definition.py
+++ b/magnum/tests/unit/drivers/test_template_definition.py
@@ -270,7 +270,7 @@ class AtomicK8sTemplateDefinitionTestCase(BaseTemplateDefinitionTestCase):
         etcd_volume_size = mock_cluster.labels.get(
             'etcd_volume_size')
         kube_tag = mock_cluster.labels.get('kube_tag')
-        container_infra_prefix = mock_cluster_template.labels.get(
+        container_infra_prefix = mock_cluster.labels.get(
             'container_infra_prefix')
 
         k8s_def = k8sa_tdef.AtomicK8sTemplateDefinition()
@@ -353,7 +353,7 @@ class AtomicK8sTemplateDefinitionTestCase(BaseTemplateDefinitionTestCase):
         etcd_volume_size = mock_cluster.labels.get(
             'etcd_volume_size')
         kube_tag = mock_cluster.labels.get('kube_tag')
-        container_infra_prefix = mock_cluster_template.labels.get(
+        container_infra_prefix = mock_cluster.labels.get(
             'container_infra_prefix')
 
         k8s_def = k8sa_tdef.AtomicK8sTemplateDefinition()