diff --git a/sdks/python/apache_beam/io/gcp/bigquery_write_it_test.py b/sdks/python/apache_beam/io/gcp/bigquery_write_it_test.py
index bb5c36a3e9b9..cd3edf19de5f 100644
--- a/sdks/python/apache_beam/io/gcp/bigquery_write_it_test.py
+++ b/sdks/python/apache_beam/io/gcp/bigquery_write_it_test.py
@@ -470,7 +470,7 @@ def test_big_query_write_insert_non_transient_api_call_error(self):
     input_data = [{
         'number': 1,
         'str': 'some_string',
-    }]*500
+    }]
 
     table_schema = {
         "fields": [{
@@ -483,7 +483,7 @@ def test_big_query_write_insert_non_transient_api_call_error(self):
     bq_result_errors = [({
         'number': 1,
         'str': 'some_string',
-    }, "Not Found")]*500
+    }, "Not Found")]
 
     args = self.test_pipeline.get_full_options_as_args()
 
@@ -491,6 +491,7 @@ def test_big_query_write_insert_non_transient_api_call_error(self):
       # pylint: disable=expression-not-assigned
       errors = (
           p | 'create' >> beam.Create(input_data)
+          | beam.WindowInto(beam.transforms.window.FixedWindows(10))
           | 'write' >> beam.io.WriteToBigQuery(
               table_id,
               schema=table_schema,