diff --git a/stages/org.osbuild.waagent.conf b/stages/org.osbuild.waagent.conf
index 69cadac8..cf7ec026 100755
--- a/stages/org.osbuild.waagent.conf
+++ b/stages/org.osbuild.waagent.conf
@@ -12,9 +12,9 @@ def bool_to_y_n(b):
 
 
 def main(tree, options):
-    sshd_config = options.get("config", {})
-    resource_disk_format = sshd_config.get("ResourceDisk.Format")
-    resource_disk_enable_swap = sshd_config.get("ResourceDisk.EnableSwap")
+    waagent_config = options.get("config", {})
+    resource_disk_format = waagent_config.get("ResourceDisk.Format")
+    resource_disk_enable_swap = waagent_config.get("ResourceDisk.EnableSwap")
     changes = {}
     if resource_disk_format is not None:
         changes["resourcedisk.format"] = {