diff --git a/builder/azure/arm/step_snapshot_data_disks_test.go b/builder/azure/arm/step_snapshot_data_disks_test.go
index 0eefa924d..aee7523ec 100644
--- a/builder/azure/arm/step_snapshot_data_disks_test.go
+++ b/builder/azure/arm/step_snapshot_data_disks_test.go
@@ -53,7 +53,7 @@ func TestStepSnapshotDataDisksShouldPassIfSnapshotPasses(t *testing.T) {
 func createTestStateBagStepSnapshotDataDisks() multistep.StateBag {
 	stateBag := new(multistep.BasicStateBag)
 
-	stateBag.Put(constants.ArmResourceGroupName, "Unit Test: ResourceGroupName")
+	stateBag.Put(constants.ArmManagedImageResourceGroupName, "Unit Test: ResourceGroupName")
 	stateBag.Put(constants.ArmLocation, "Unit Test: Location")
 
 	value := "Unit Test: Tags"
diff --git a/builder/azure/arm/step_snapshot_os_disk_test.go b/builder/azure/arm/step_snapshot_os_disk_test.go
index 786e6a36e..f4e7eeafb 100644
--- a/builder/azure/arm/step_snapshot_os_disk_test.go
+++ b/builder/azure/arm/step_snapshot_os_disk_test.go
@@ -53,7 +53,7 @@ func TestStepSnapshotOSDiskShouldPassIfSnapshotPasses(t *testing.T) {
 func createTestStateBagStepSnapshotOSDisk() multistep.StateBag {
 	stateBag := new(multistep.BasicStateBag)
 
-	stateBag.Put(constants.ArmResourceGroupName, "Unit Test: ResourceGroupName")
+	stateBag.Put(constants.ArmManagedImageResourceGroupName, "Unit Test: ResourceGroupName")
 	stateBag.Put(constants.ArmLocation, "Unit Test: Location")
 
 	value := "Unit Test: Tags"