diff --git a/classes/cluster/os-ha-contrail-40/infra/config.yml b/classes/cluster/os-ha-contrail-40/infra/config.yml
index b93074a..97f1070 100644
--- a/classes/cluster/os-ha-contrail-40/infra/config.yml
+++ b/classes/cluster/os-ha-contrail-40/infra/config.yml
@@ -94,6 +94,11 @@
               value_template: <<node_os>>
             salt_master_host:
               value_template: <<node_master_ip>>
+        mark_contrail_version:
+          expression: <<opencontrail_version>>__startswith__oc
+          cluster_param:
+            linux_repo_contrail_component:
+              value_template: <<opencontrail_version>>
         infra_config:
           expression: <<node_hostname>>__startswith__cfg
           cluster_param: