diff --git a/tap_cassandra/streams.py b/tap_cassandra/streams.py index 1a06264..f11cbe3 100644 --- a/tap_cassandra/streams.py +++ b/tap_cassandra/streams.py @@ -36,8 +36,8 @@ def get_records(self, context): cql = f"select {selected_column_string} from {self.name.split('-')[1]}" if self.config.get('skip_hot_partitions'): - for row in self.execute_with_skip(cql, self.catalog_entry['key_properties'][0]): + for row in self.connector.execute_with_skip(cql, self.catalog_entry['key_properties'][0]): yield row else: - for row in self.execute(cql): + for row in self.connector.execute(cql): yield row