diff --git a/tljh/config_schema.py b/tljh/config_schema.py
index e6fb820..9d141c6 100644
--- a/tljh/config_schema.py
+++ b/tljh/config_schema.py
@@ -79,23 +79,16 @@ config_schema = {
             "description": "User CPU and memory limits.",
             "type": "object",
             "additionalProperties": False,
-            "properties": {"memory": {"type": "string"}, "cpu": {
-            "oneOf": [
-                {
-                  "type": "integer",
-                  "minimum": 0
+            "properties": {
+                "memory": {"type": "string"},
+                "cpu": {
+                    "anyOf": [
+                        {"type": "integer", "minimum": 0},
+                        {"type": "number", "minimum": 0},
+                        {"type": "string", "enum": ["None"]},
+                    ]
                 },
-                {
-                  "type": "number",
-                  "minimum": 0
-                },
-                {
-                  "type": "string",
-                  "enum": ["None"]
-                }
-            ]
-          }
-        },
+            },
         },
         "UserEnvironment": {
             "type": "object",