diff --git a/app/Filament/Admin/Pages/Waf.php b/app/Filament/Admin/Pages/Waf.php
index 02c6872..9d0545d 100644
--- a/app/Filament/Admin/Pages/Waf.php
+++ b/app/Filament/Admin/Pages/Waf.php
@@ -408,7 +408,8 @@ class Waf extends Page implements HasForms, HasTable
                 ->send();
         }
 
-        $this->auditEntries = $this->markWhitelisted($this->auditEntries);
+        $this->loadAuditLogs(false);
+        $this->resetTable();
 
         Notification::make()
             ->title(__('Rule whitelisted'))
@@ -452,7 +453,8 @@ class Waf extends Page implements HasForms, HasTable
                 ->send();
         }
 
-        $this->auditEntries = $this->markWhitelisted($this->auditEntries);
+        $this->loadAuditLogs(false);
+        $this->resetTable();
 
         Notification::make()
             ->title(__('Whitelist removed'))