From bf07eafa633baae77ee901b9a290749a0ef82409 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Mindaugas=20Ni=C5=BEauskas?= <mindaugas@nizauskas.com>
Date: Mon, 8 Jan 2024 11:27:00 +0200
Subject: [PATCH] add connector

---
 tap_cassandra/streams.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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