diff --git a/stages/org.osbuild.cloud-init b/stages/org.osbuild.cloud-init
index 1df21fbf..1d5da2a7 100755
--- a/stages/org.osbuild.cloud-init
+++ b/stages/org.osbuild.cloud-init
@@ -57,6 +57,31 @@ SCHEMA = r"""
             }
           }
         }
+      },
+      "datasource_list": {
+        "type": "array",
+        "items": {
+          "type": "string",
+          "enum": ["Azure"]
+        }
+      },
+      "datasource": {
+        "type": "object",
+        "description": "Sources of configuration data for cloud-init.",
+        "minProperties": 1,
+        "properties": {
+          "Azure": {
+            "type": "object",
+            "minProperties": 1,
+            "properties": {
+              "apply_network_config": {
+                "type": "boolean",
+                "description": "Whether to use network configuration described by Azure’s IMDS endpoint",
+                "default": true
+              }
+            }
+          }
+        }
       }
     }
   }