diff --git a/cli/internal/cmd/init_test.go b/cli/internal/cmd/init_test.go
index 7b401fd05..4cf643325 100644
--- a/cli/internal/cmd/init_test.go
+++ b/cli/internal/cmd/init_test.go
@@ -606,6 +606,7 @@ func defaultConfigWithExpectedMeasurements(t *testing.T, conf *config.Config, cs
 	conf.Image = constants.BinaryVersion().String()
 	conf.Name = "kubernetes"
 
+	var zone, instanceType, diskType string
 	switch csp {
 	case cloudprovider.Azure:
 		conf.Provider.Azure.SubscriptionID = "01234567-0123-0123-0123-0123456789ab"
@@ -616,6 +617,8 @@ func defaultConfigWithExpectedMeasurements(t *testing.T, conf *config.Config, cs
 		conf.Attestation.AzureSEVSNP.Measurements[4] = measurements.WithAllBytes(0x44, measurements.Enforce, measurements.PCRMeasurementLength)
 		conf.Attestation.AzureSEVSNP.Measurements[9] = measurements.WithAllBytes(0x11, measurements.Enforce, measurements.PCRMeasurementLength)
 		conf.Attestation.AzureSEVSNP.Measurements[12] = measurements.WithAllBytes(0xcc, measurements.Enforce, measurements.PCRMeasurementLength)
+		instanceType = "Standard_DC4as_v5"
+		diskType = "StandardSSD_LRS"
 	case cloudprovider.GCP:
 		conf.Provider.GCP.Region = "test-region"
 		conf.Provider.GCP.Project = "test-project"
@@ -624,13 +627,22 @@ func defaultConfigWithExpectedMeasurements(t *testing.T, conf *config.Config, cs
 		conf.Attestation.GCPSEVES.Measurements[4] = measurements.WithAllBytes(0x44, measurements.Enforce, measurements.PCRMeasurementLength)
 		conf.Attestation.GCPSEVES.Measurements[9] = measurements.WithAllBytes(0x11, measurements.Enforce, measurements.PCRMeasurementLength)
 		conf.Attestation.GCPSEVES.Measurements[12] = measurements.WithAllBytes(0xcc, measurements.Enforce, measurements.PCRMeasurementLength)
+		zone = "europe-west3-b"
+		instanceType = "n2d-standard-4"
+		diskType = "pd-ssd"
 	case cloudprovider.QEMU:
 		conf.Attestation.QEMUVTPM.Measurements[4] = measurements.WithAllBytes(0x44, measurements.Enforce, measurements.PCRMeasurementLength)
 		conf.Attestation.QEMUVTPM.Measurements[9] = measurements.WithAllBytes(0x11, measurements.Enforce, measurements.PCRMeasurementLength)
 		conf.Attestation.QEMUVTPM.Measurements[12] = measurements.WithAllBytes(0xcc, measurements.Enforce, measurements.PCRMeasurementLength)
 	}
 
-	conf.RemoveProviderAndAttestationExcept(csp)
+	for groupName, group := range conf.NodeGroups {
+		group.Zone = zone
+		group.InstanceType = instanceType
+		group.StateDiskType = diskType
+		conf.NodeGroups[groupName] = group
+	}
+
 	return conf
 }