diff --git a/builder/azure/arm/config.hcl2spec.go b/builder/azure/arm/config.hcl2spec.go
index ce473b0f..b8b28306 100644
--- a/builder/azure/arm/config.hcl2spec.go
+++ b/builder/azure/arm/config.hcl2spec.go
@@ -39,7 +39,7 @@ type FlatConfig struct {
 	SharedGalleryDestination                   *FlatSharedImageGalleryDestination `mapstructure:"shared_image_gallery_destination" cty:"shared_image_gallery_destination" hcl:"shared_image_gallery_destination"`
 	SharedGalleryTimeout                       *string                            `mapstructure:"shared_image_gallery_timeout" cty:"shared_image_gallery_timeout" hcl:"shared_image_gallery_timeout"`
 	SharedGalleryImageVersionEndOfLifeDate     *string                            `mapstructure:"shared_gallery_image_version_end_of_life_date" required:"false" cty:"shared_gallery_image_version_end_of_life_date" hcl:"shared_gallery_image_version_end_of_life_date"`
-	SharedGalleryImageVersionReplicaCount      *int32                             `mapstructure:"shared_image_gallery_replica_count" required:"false" cty:"shared_image_gallery_replica_count" hcl:"shared_image_gallery_replica_count"`
+	SharedGalleryImageVersionReplicaCount      *int64                             `mapstructure:"shared_image_gallery_replica_count" required:"false" cty:"shared_image_gallery_replica_count" hcl:"shared_image_gallery_replica_count"`
 	SharedGalleryImageVersionExcludeFromLatest *bool                              `mapstructure:"shared_gallery_image_version_exclude_from_latest" required:"false" cty:"shared_gallery_image_version_exclude_from_latest" hcl:"shared_gallery_image_version_exclude_from_latest"`
 	ImagePublisher                             *string                            `mapstructure:"image_publisher" required:"true" cty:"image_publisher" hcl:"image_publisher"`
 	ImageOffer                                 *string                            `mapstructure:"image_offer" required:"true" cty:"image_offer" hcl:"image_offer"`
@@ -367,7 +367,7 @@ type FlatSharedImageGalleryDestination struct {
 	SigDestinationReplicationRegions     []string `mapstructure:"replication_regions" cty:"replication_regions" hcl:"replication_regions"`
 	SigDestinationStorageAccountType     *string  `mapstructure:"storage_account_type" cty:"storage_account_type" hcl:"storage_account_type"`
 	SigDestinationSpecialized            *bool    `mapstructure:"specialized" cty:"specialized" hcl:"specialized"`
-	SigDestinationShallowReplicationMode *bool    `mapstructure:"shallow_replication" cty:"shallow_replication" hcl:"shallow_replication"`
+	SigDestinationShallowReplicationMode *bool    `mapstructure:"shallow_replication" required:"false" cty:"shallow_replication" hcl:"shallow_replication"`
 }
 
 // FlatMapstructure returns a new FlatSharedImageGalleryDestination.
diff --git a/builder/azure/arm/step_publish_to_shared_image_gallery_test.go b/builder/azure/arm/step_publish_to_shared_image_gallery_test.go
index 627ba98d..cfa7945e 100644
--- a/builder/azure/arm/step_publish_to_shared_image_gallery_test.go
+++ b/builder/azure/arm/step_publish_to_shared_image_gallery_test.go
@@ -104,6 +104,7 @@ func createTestStateBagStepPublishToSharedImageGallery(managed bool) multistep.S
 	stateBag.Put(constants.ArmSharedImageGalleryDestinationSubscription, "Unit Test: ManagedImageSubscription")
 	stateBag.Put(constants.ArmIsManagedImage, managed)
 	stateBag.Put(constants.ArmIsSIGImage, true)
+	stateBag.Put(constants.ArmSharedImageGalleryDestinationShallowReplication, false)
 
 	return stateBag
 }
diff --git a/docs-partials/builder/azure/arm/Config-not-required.mdx b/docs-partials/builder/azure/arm/Config-not-required.mdx
index 82f14e40..440667b1 100644
--- a/docs-partials/builder/azure/arm/Config-not-required.mdx
+++ b/docs-partials/builder/azure/arm/Config-not-required.mdx
@@ -86,7 +86,7 @@
 - `shared_gallery_image_version_end_of_life_date` (string) - The end of life date (2006-01-02T15:04:05.99Z) of the gallery Image Version. This property
   can be used for decommissioning purposes.
 
-- `shared_image_gallery_replica_count` (int32) - The number of replicas of the Image Version to be created per region. This
+- `shared_image_gallery_replica_count` (int64) - The number of replicas of the Image Version to be created per region. This
   property would take effect for a region when regionalReplicaCount is not specified.
   Replica count must be between 1 and 100, but 50 replicas should be sufficient for most use cases.