use pollingContext for data disk snapshot (#419)

Co-authored-by: dreweasland <drew_easland@trekbikes.com>
hashicorp · May 28, 2024 · 182e0b6 · 182e0b6
1 parent 5e1cb75
commit 182e0b6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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