From 067f7a7e27aee94971dbf28641883155246482a2 Mon Sep 17 00:00:00 2001
From: Piotr Szczepanik <piotr.szczepanik@allegro.pl>
Date: Wed, 29 Sep 2021 20:43:44 +0200
Subject: [PATCH] [BEAM-12356] Fixed last non-cached usage of DatasetService in
 BigQuery WriteFiles

---
 .../java/org/apache/beam/sdk/io/gcp/bigquery/WriteTables.java   | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/sdks/java/io/google-cloud-platform/src/main/java/org/apache/beam/sdk/io/gcp/bigquery/WriteTables.java b/sdks/java/io/google-cloud-platform/src/main/java/org/apache/beam/sdk/io/gcp/bigquery/WriteTables.java
index 32ed1fe61275..2a2697ac2fb1 100644
--- a/sdks/java/io/google-cloud-platform/src/main/java/org/apache/beam/sdk/io/gcp/bigquery/WriteTables.java
+++ b/sdks/java/io/google-cloud-platform/src/main/java/org/apache/beam/sdk/io/gcp/bigquery/WriteTables.java
@@ -332,7 +332,7 @@ public void populateDisplayData(DisplayData.Builder builder) {
     @FinishBundle
     public void finishBundle(FinishBundleContext c) throws Exception {
       DatasetService datasetService =
-          bqServices.getDatasetService(c.getPipelineOptions().as(BigQueryOptions.class));
+          getDatasetService(c.getPipelineOptions().as(BigQueryOptions.class));
 
       PendingJobManager jobManager = new PendingJobManager();
       for (final PendingJobData pendingJob : pendingJobs) {