diff --git a/tests/pillar/master_contrail.sls b/tests/pillar/master_contrail.sls
index 39f9956..c237389 100644
--- a/tests/pillar/master_contrail.sls
+++ b/tests/pillar/master_contrail.sls
@@ -34,9 +34,10 @@
         enabled: true
         namespace: kube-system
         image: image
-      kube_network_manager:
+      contrail_network_controller:
         enabled: true
         namespace: kube-system
+        image: yashulyak/contrail-controller:latest
     admin:
       password: password
       username: admin
diff --git a/tests/pillar/master_contrail4_0.sls b/tests/pillar/master_contrail4_0.sls
index b687840..8cf9c36 100644
--- a/tests/pillar/master_contrail4_0.sls
+++ b/tests/pillar/master_contrail4_0.sls
@@ -31,9 +31,10 @@
         server_image: image
         agent_image: image
         agent_probeurls: "http://ipinfo.io"
-      kube_network_manager:
+      contrail_network_controller:
         enabled: true
         namespace: kube-system
+        image: yashulyak/contrail-controller:latest
     admin:
       password: password
       username: admin