diff --git a/src/lib/testFlags.ts b/src/lib/testFlags.ts
index 3a2acb1fa..04d24ca95 100644
--- a/src/lib/testFlags.ts
+++ b/src/lib/testFlags.ts
@@ -1,3 +1,5 @@
+import { isDev } from '@/constants/networks';
+
 class TestFlags {
   public queryParams: { [key: string]: string };
 
@@ -32,16 +34,8 @@ class TestFlags {
     return this.queryParams.address;
   }
 
-  get isolatedMargin() {
-    return !!this.queryParams.isolatedmargin;
-  }
-
-  get enableComplianceApi() {
-    return !!this.queryParams.complianceapi;
-  }
-
   get enableVaults() {
-    return !!this.queryParams.vaults || import.meta.env.MODE === 'staging';
+    return !!this.queryParams.vaults || isDev;
   }
 
   get referrer() {
@@ -49,11 +43,11 @@ class TestFlags {
   }
 
   get enablePredictionMarketPerp() {
-    return !!this.queryParams.prediction || import.meta.env.MODE === 'staging';
+    return !!this.queryParams.prediction || isDev;
   }
 
   get pml() {
-    return !!this.queryParams.pml;
+    return !!this.queryParams.pml || isDev;
   }
 
   get showLimitClose() {
@@ -69,7 +63,7 @@ class TestFlags {
   }
 
   get uiRefresh() {
-    return !!this.queryParams.uirefresh;
+    return !!this.queryParams.uirefresh || isDev;
   }
 }