From e28cb3036655dc3c944a210264aa08f98e55e6dc Mon Sep 17 00:00:00 2001
From: Ilija Lazoroski <ilija.la@live.com>
Date: Fri, 1 Jul 2022 14:38:55 +0200
Subject: [PATCH] UI: Add custom pbas and pbas to the UiSchema

---
 .../configuration-components/UiSchema.js           | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

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 a4034e35e..2f37ae84e 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
@@ -28,15 +28,23 @@ export default function UiSchema(props) {
         }
       }
     },
-    credential_collectors: {
-      collectors: {
+    custom_pbas : {
+      classNames: 'config-template-no-header'
+    },
+    post_breach_actions: {
         classNames: 'config-template-no-header',
         'ui:widget': AdvancedMultiSelect,
+        post_breach_actions: {
+          classNames: 'config-template-no-header'
+        }
+    },
+    credential_collectors: {
+      classNames: 'config-template-no-header',
+      'ui:widget': AdvancedMultiSelect,
         credential_collectors_classes :{
           classNames: 'config-template-no-header'
         }
       }
-    }
   };
   return UiSchema[props.selectedSection]
 }