diff --git a/core/src/epicli/tests/cli/engine/providers/aws/test_AWSConfigBuilder.py b/core/src/epicli/tests/cli/engine/providers/aws/test_AWSConfigBuilder.py
index e3fd23f37c..98fd8bb456 100644
--- a/core/src/epicli/tests/cli/engine/providers/aws/test_AWSConfigBuilder.py
+++ b/core/src/epicli/tests/cli/engine/providers/aws/test_AWSConfigBuilder.py
@@ -111,7 +111,10 @@ def get_cluster_model(address_pool='10.22.0.0/22', cluster_name='EpiphanyTestClu
             'name': cluster_name,
             'prefix': 'prefix',
             'cloud': {
-                'vnet_address_pool': address_pool
+                'vnet_address_pool': address_pool,
+                'network': {
+                    'use_network_security_groups': True
+                }
             }
         }
     })
diff --git a/core/src/epicli/tests/cli/engine/providers/azure/test_AzureConfigBuilder.py b/core/src/epicli/tests/cli/engine/providers/azure/test_AzureConfigBuilder.py
index 6663db0bda..3b307956bb 100644
--- a/core/src/epicli/tests/cli/engine/providers/azure/test_AzureConfigBuilder.py
+++ b/core/src/epicli/tests/cli/engine/providers/azure/test_AzureConfigBuilder.py
@@ -125,7 +125,10 @@ def get_cluster_model(address_pool='10.22.0.0/22', cluster_name='EpiphanyTestClu
             'cloud': {
                 'region': 'West Europe',
                 'vnet_address_pool': address_pool,
-                'use_public_ips': True
+                'use_public_ips': True,
+                'network': {
+                    'use_network_security_groups': True
+                }
             }
         }
     })