diff --git a/frontend/src/app/modules/common/step-tree/step-tree.component.ts b/frontend/src/app/modules/common/step-tree/step-tree.component.ts
index 0790bf2fb9..f000bd1a3f 100644
--- a/frontend/src/app/modules/common/step-tree/step-tree.component.ts
+++ b/frontend/src/app/modules/common/step-tree/step-tree.component.ts
@@ -21,7 +21,7 @@ interface TreeNode {
 export class StepTreeComponent {
     treeControl = new NestedTreeControl<TreeNode>((node) => node.children);
     // dataSource = new MatTreeNestedDataSource<TreeNode>();
-    dataSource: any
+    dataSource: { data: TreeNode[] } = { data: [] }
 
     @Input('treeData') treeData!: any;
     @Input('currentNode') currentNode!: any;
diff --git a/frontend/src/app/modules/policy-engine/dialogs/policy-wizard-dialog/policy-wizard-dialog.component.html b/frontend/src/app/modules/policy-engine/dialogs/policy-wizard-dialog/policy-wizard-dialog.component.html
index 2917eeeb8b..b25bf25126 100644
--- a/frontend/src/app/modules/policy-engine/dialogs/policy-wizard-dialog/policy-wizard-dialog.component.html
+++ b/frontend/src/app/modules/policy-engine/dialogs/policy-wizard-dialog/policy-wizard-dialog.component.html
@@ -50,6 +50,9 @@
                     optionValue="id"
                     placeholder="Not selected"
             >
+                <ng-template pTemplate="selectedItem" let-selectedItem>
+                    {{ selectedItem ? selectedItem.name : 'Not selected' }}
+                </ng-template>
             </p-dropdown>
         </div>
         <div class="form-input-container">
@@ -63,6 +66,9 @@
                     optionValue="id"
                     placeholder="Not selected"
             >
+                <ng-template pTemplate="selectedItem" let-selectedItem>
+                    {{ selectedItem ? selectedItem.name : 'Not selected' }}
+                </ng-template>
             </p-dropdown>
         </div>
         <div class="form-input-container">
@@ -118,6 +124,9 @@ <h3>Categorization</h3>
                     optionValue="id"
                     placeholder="Not selected"
             >
+                <ng-template pTemplate="selectedItem" let-selectedItem>
+                    {{ selectedItem ? selectedItem.name : 'Not selected' }}
+                </ng-template>
             </p-dropdown>
         </div>
         <div class="form-input-container">