diff --git a/builder/azure/chroot/step_resolve_plaform_image_version_test.go b/builder/azure/chroot/step_resolve_plaform_image_version_test.go
index 695a1a149..77210b1a9 100644
--- a/builder/azure/chroot/step_resolve_plaform_image_version_test.go
+++ b/builder/azure/chroot/step_resolve_plaform_image_version_test.go
@@ -39,8 +39,8 @@ func TestStepResolvePlatformImageVersion_Run(t *testing.T) {
 
 	state := new(multistep.BasicStateBag)
 	state.Put("azureclient", &client.AzureClientSetMock{
-		VirtualMachineImagesClientMock: client.VirtualMachineImagesClient{m},
-	})
+		VirtualMachineImagesClientMock: client.VirtualMachineImagesClient{
+			VirtualMachineImagesClientAPI: m}})
 
 	ui, getErrs := testUI()
 	state.Put("ui", ui)