From aa6e9255fe5a5b906bcad470036554b6d88f69de Mon Sep 17 00:00:00 2001
From: Malte Poll <mp@edgeless.systems>
Date: Wed, 2 Aug 2023 10:42:18 +0200
Subject: [PATCH] cli: use nodeGroups when setting default disk size for QEMU

---
 cli/internal/cmd/configgenerate.go | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/cli/internal/cmd/configgenerate.go b/cli/internal/cmd/configgenerate.go
index d982a760b..02fd0fb4d 100644
--- a/cli/internal/cmd/configgenerate.go
+++ b/cli/internal/cmd/configgenerate.go
@@ -108,7 +108,10 @@ func createConfigWithAttestationVariant(provider cloudprovider.Provider, rawProv
 
 	// set a lower default for QEMU's state disk
 	if provider == cloudprovider.QEMU {
-		conf.StateDiskSizeGB = 10
+		for groupName, group := range conf.NodeGroups {
+			group.StateDiskSizeGB = 10
+			conf.NodeGroups[groupName] = group
+		}
 	}
 
 	if provider == cloudprovider.Unknown {