From 7dfac1f758b80f2ab15a931c0f9e9573478b528d 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 {