--- a/browser/extensions/doh-rollout/experiments/preferences/schema.json
+++ b/browser/extensions/doh-rollout/experiments/preferences/schema.json
@@ -19,17 +19,18 @@
     "functions": [
       {
         "name": "getIntPref",
         "type": "function",
         "description": "Get the value of a integer preference",
         "parameters": [
           {
             "type": "string",
-            "name": "name"
+            "name": "name",
+            "enum": ["doh-rollout.previous.trr.mode"]
           },
           {
             "type": "integer",
             "name": "defaultValue"
           }
         ],
         "async": true
       },
@@ -51,49 +52,72 @@
       },
       {
         "name": "getBoolPref",
         "type": "function",
         "description": "Get the value of a boolean preference",
         "parameters": [
           {
             "type": "string",
-            "name": "name"
+            "name": "name",
+            "enum": [
+              "doh-rollout.enabled",
+              "doh-rollout.self-enabled",
+              "doh-rollout.doorhanger-shown",
+              "doh-rollout.disable-heuristics",
+              "doh-rollout.skipHeuristicsCheck",
+              "doh-rollout.doneFirstRun",
+              "doh-rollout.balrog-migration-done",
+              "doh-rollout.debug",
+              "doh-rollout.provider-steering.enabled",
+              "security.enterprise_roots.enabled"
+            ]
           },
           {
             "type": "boolean",
             "name": "defaultValue"
           }
         ],
         "async": true
       },
       {
         "name": "setBoolPref",
         "type": "function",
         "description": "Sets the value of a boolean preference",
         "parameters": [
           {
             "type": "string",
-            "enum": ["doh-rollout.doorhanger-shown", "doh-rollout.self-enabled", "doh-rollout.disable-heuristics", "doh-rollout.doneFirstRun", "doh-rollout.skipHeuristicsCheck", "doh-rollout.balrog-migration-done"]
+            "enum": [
+              "doh-rollout.doorhanger-shown",
+              "doh-rollout.self-enabled",
+              "doh-rollout.disable-heuristics",
+              "doh-rollout.doneFirstRun",
+              "doh-rollout.skipHeuristicsCheck",
+              "doh-rollout.balrog-migration-done"
+            ]
           },
           {
             "type": "boolean",
             "name": "defaultValue"
           }
         ],
         "async": true
       },
       {
         "name": "getCharPref",
         "type": "function",
         "description": "Gets the value of a string preference",
         "parameters": [
           {
             "type": "string",
-            "enum": ["doh-rollout.doorhanger-decision", "doh-rollout.heuristics.mockValues", "doh-rollout.provider-steering.provider-list"]
+            "enum": [
+              "doh-rollout.doorhanger-decision",
+              "doh-rollout.heuristics.mockValues",
+              "doh-rollout.provider-steering.provider-list"
+            ]
           },
           {
             "type": "string",
             "name": "defaultValue"
           }
         ],
         "async": true
       },
@@ -115,29 +139,35 @@
       },
       {
         "name": "clearUserPref",
         "type": "function",
         "description": "Resets value of prefence back to default",
         "parameters": [
           {
             "type": "string",
-            "enum": ["doh-rollout.self-enabled", "doh-rollout.mode", "network.trr.mode", "doh-rollout.previous.trr.mode"]
+            "enum": [
+              "doh-rollout.self-enabled",
+              "doh-rollout.mode",
+              "network.trr.mode",
+              "doh-rollout.previous.trr.mode"
+            ]
           }
         ],
         "async": true
       },
       {
         "name": "prefHasUserValue",
         "type": "function",
         "description": "Check if the user has set a value of a preference",
         "parameters": [
           {
             "type": "string",
-            "name": "name"
+            "name": "name",
+            "enum": ["network.trr.mode", "network.trr.uri"]
           }
         ],
         "async": true
       },
       {
         "name": "migrateNextDNSEndpoint",
         "type": "function",
         "description": "Migrates any occurrances of old NextDNS endpoint URL in pref values to the new endpoint.",