diff --git a/ui/src/app/pages/datasets/policies.agent/view/agent.policy.view.component.ts b/ui/src/app/pages/datasets/policies.agent/view/agent.policy.view.component.ts
index bc5e0baa3..a90bb6bc3 100644
--- a/ui/src/app/pages/datasets/policies.agent/view/agent.policy.view.component.ts
+++ b/ui/src/app/pages/datasets/policies.agent/view/agent.policy.view.component.ts
@@ -148,7 +148,7 @@ export class AgentPolicyViewComponent implements OnInit, OnDestroy, OnChanges {
       this.policiesService.editAgentPolicy(payload).subscribe((resp) => {
         this.discard();
         this.policy = resp;
-        this.cdr.markForCheck();
+        this.fetchData();
       });
 
       this.notifications.success('Agent Policy updated successfully', '');