diff --git a/themes/schema.json b/themes/schema.json
index 5052637c..945cf0a4 100644
--- a/themes/schema.json
+++ b/themes/schema.json
@@ -21,7 +21,7 @@
       "type": "string",
       "title": "Display Mode",
       "description": "Determines whether the segment is displayed always or only if a file matching the extensions are present in the current folder",
-      "enum": ["always", "files"],
+      "enum": ["always", "files", "environment", "context"],
       "default": "context"
     },
     "missing_command_text": {