diff --git a/monkey/monkey_island/cc/ui/src/components/configuration-components/UiSchema.js b/monkey/monkey_island/cc/ui/src/components/configuration-components/UiSchema.js
index bbc55a73e..20d5ccc84 100644
--- a/monkey/monkey_island/cc/ui/src/components/configuration-components/UiSchema.js
+++ b/monkey/monkey_island/cc/ui/src/components/configuration-components/UiSchema.js
@@ -4,13 +4,18 @@ import InfoBox from './InfoBox';
 export default function UiSchema(props) {
   const UiSchema = {
     propagation: {
-      'ui:order': ['exploitation', 'maximum_depth', 'network_scan'],
       exploitation: {
         brute_force: {
-          'ui:widget': AdvancedMultiSelect
+          brute_force_classes: {
+            classNames: 'config-template-no-header',
+            'ui:widget': AdvancedMultiSelect
+          }
         },
         vulnerability: {
-          'ui:widget': AdvancedMultiSelect
+          vulnerability_classes: {
+            classNames: 'config-template-no-header',
+            'ui:widget': AdvancedMultiSelect
+          }
         }
       },
       network_scan: {