diff --git a/builder/azure/arm/step_snapshot_data_disks.go b/builder/azure/arm/step_snapshot_data_disks.go
index 93ce63d9..358d6ba0 100644
--- a/builder/azure/arm/step_snapshot_data_disks.go
+++ b/builder/azure/arm/step_snapshot_data_disks.go
@@ -56,7 +56,7 @@ func (s *StepSnapshotDataDisks) createDataDiskSnapshot(ctx context.Context, subs
 		return err
 	}
 
-	snapshot, err := s.client.SnapshotsClient.Get(ctx, id)
+	snapshot, err := s.client.SnapshotsClient.Get(pollingContext, id)
 	if err != nil {
 		s.say(s.client.LastError.Error())
 		return err