diff --git a/RemoveWindowsAi.ps1 b/RemoveWindowsAi.ps1
index dcd0029..1cb88b6 100644
--- a/RemoveWindowsAi.ps1
+++ b/RemoveWindowsAi.ps1
@@ -283,9 +283,14 @@ function Disable-Registry-Keys {
             'edge-ntp-composer@2', #disables the copilot search in new tab page 
             'edge-compose@2' #disables the ai writing help 
         )
-        foreach ($flag in $flags) {
-            if ($jsonContent.browser.enabled_labs_experiments -notcontains $flag) {
-                $jsonContent.browser.enabled_labs_experiments += $flag
+        if ($revert) {
+            $jsonContent.browser.enabled_labs_experiments = $jsonContent.browser.enabled_labs_experiments | Where-Object { $_ -notin $flags }
+        }
+        else {
+            foreach ($flag in $flags) {
+                if ($jsonContent.browser.enabled_labs_experiments -notcontains $flag) {
+                    $jsonContent.browser.enabled_labs_experiments += $flag
+                }
             }
         }