diff --git a/sdks/python/apache_beam/io/gcp/bigquery.py b/sdks/python/apache_beam/io/gcp/bigquery.py index 45461eccc7f7..428e2ae4885a 100644 --- a/sdks/python/apache_beam/io/gcp/bigquery.py +++ b/sdks/python/apache_beam/io/gcp/bigquery.py @@ -2751,7 +2751,7 @@ def with_output_types(self): fields = [(StorageWriteToBigQuery.DESTINATION, str), (StorageWriteToBigQuery.RECORD, self.element_type)] if callable(self.mutation_info_fn): - fields.append(CDC_INFO_TYPE_HINT) + fields.append(StorageWriteToBigQuery.CDC_INFO_TYPE_HINT) type_hint = RowTypeConstraint.from_fields(fields) else: if callable(self.mutation_info_fn): @@ -2812,7 +2812,7 @@ def with_output_types(self): StorageWriteToBigQuery.RECORD, RowTypeConstraint.from_fields(row_type_hints))] if callable(self.mutation_info_fn): - fields.append(CDC_INFO_TYPE_HINT) + fields.append(StorageWriteToBigQuery.CDC_INFO_TYPE_HINT) type_hint = RowTypeConstraint.from_fields(fields) else: if callable(self.mutation_info_fn):