diff --git a/testutils/resourcebuilder/machine/v1beta1/azure_provider_spec.go b/testutils/resourcebuilder/machine/v1beta1/azure_provider_spec.go
index f2057cbf8..a578bcaa2 100644
--- a/testutils/resourcebuilder/machine/v1beta1/azure_provider_spec.go
+++ b/testutils/resourcebuilder/machine/v1beta1/azure_provider_spec.go
@@ -41,6 +41,7 @@ type AzureProviderSpecBuilder struct {
 	vmSize               string
 	zone                 string
 	subnet               string
+	image                machinev1beta1.Image
 }
 
 // Build builds a new Azure machine config based on the configuration provided.
@@ -120,3 +121,9 @@ func (m AzureProviderSpecBuilder) WithSubnet(subnet string) AzureProviderSpecBui
 	m.subnet = subnet
 	return m
 }
+
+// WithImage sets the image for the Azure machine config builder.
+func (m AzureProviderSpecBuilder) WithImage(image machinev1beta1.Image) AzureProviderSpecBuilder {
+	m.image = image
+	return m
+}