diff --git a/tests/test_flatten_ga_data.py b/tests/test_flatten_ga_data.py
index 0e693a5..0c5e5b0 100644
--- a/tests/test_flatten_ga_data.py
+++ b/tests/test_flatten_ga_data.py
@@ -9,7 +9,7 @@ class TestCFFlattenMethods(BaseUnitTest):
     c = Context()
     ga_source = GaExportedNestedDataStorage(gcp_project=c.env["project"],
                                             dataset=c.env["dataset"],
-                                            table_name=c.env["table_type"],
+                                            table_type=c.env["table_type"],
                                             date_shard=c.env["date"],
                                             )
 
@@ -63,7 +63,7 @@ class TestCFFlattenMethodsSchemaChangeCollectedTrafficSource(BaseUnitTest):
     c = Context()
     ga_source = GaExportedNestedDataStorage(gcp_project=c.env["project"],
                                             dataset=c.env["dataset_adswerve"],
-                                            table_name=c.env["table_type"],
+                                            table_type=c.env["table_type"],
                                             date_shard=c.env["date_collected_traffic_source_added"],
                                             )
 
@@ -90,7 +90,7 @@ class TestCFFlattenMethodsSchemaChangeIsActiveUser(BaseUnitTest):
     c = Context()
     ga_source = GaExportedNestedDataStorage(gcp_project=c.env["project"],
                                             dataset=c.env["dataset_adswerve"],
-                                            table_name=c.env["table_type"],
+                                            table_type=c.env["table_type"],
                                             date_shard=c.env["date_is_active_user_added"],
                                             )
 
diff --git a/tests/test_flatten_ga_data_intraday.py b/tests/test_flatten_ga_data_intraday.py
index bb2be30..893be7a 100644
--- a/tests/test_flatten_ga_data_intraday.py
+++ b/tests/test_flatten_ga_data_intraday.py
@@ -9,7 +9,7 @@ class TestCFFlattenMethodsIntraday(BaseUnitTest):
     c = Context()
     ga_source = GaExportedNestedDataStorage(gcp_project=c.env["project"],
                                             dataset=c.env["dataset"],
-                                            table_name=c.env["table_type_intraday"],
+                                            table_type=c.env["table_type_intraday"],
                                             date_shard=c.env["date_intraday"],
                                             )
 
diff --git a/tests/test_partitioning.py b/tests/test_partitioning.py
index 9624972..c2163b3 100644
--- a/tests/test_partitioning.py
+++ b/tests/test_partitioning.py
@@ -20,7 +20,7 @@ class TestPartitioning(BaseUnitTest):
     c = Context()
     ga_source = GaExportedNestedDataStorage(gcp_project=c.env["project"],
                                             dataset=c.env["dataset"],
-                                            table_name=c.env["table_type"],
+                                            table_type=c.env["table_type"],
                                             date_shard=c.env["date"],
                                             )
 
diff --git a/tests/test_verify_queries.py b/tests/test_verify_queries.py
index f678cef..519201e 100644
--- a/tests/test_verify_queries.py
+++ b/tests/test_verify_queries.py
@@ -9,7 +9,7 @@ class TestGenerateQuery(BaseUnitTest):
     c = Context()
     ga_source = GaExportedNestedDataStorage(gcp_project=c.env["project"],
                                             dataset=c.env["dataset"],
-                                            table_name=c.env["table_type"],
+                                            table_type=c.env["table_type"],
                                             date_shard=c.env["date"],
                                             )
 
diff --git a/tests/test_verify_queries_schema_change.py b/tests/test_verify_queries_schema_change.py
index 892d4e6..f100ffd 100644
--- a/tests/test_verify_queries_schema_change.py
+++ b/tests/test_verify_queries_schema_change.py
@@ -9,7 +9,7 @@ class TestGenerateQuerySchemaChange1(BaseUnitTest):
     c = Context()
     ga_source = GaExportedNestedDataStorage(gcp_project=c.env["project"],
                                             dataset=c.env["dataset"],
-                                            table_name=c.env["table_type"],
+                                            table_type=c.env["table_type"],
                                             date_shard=c.env["date_collected_traffic_source_added"],
                                             )
 
@@ -65,7 +65,7 @@ class TestGenerateQuerySchemaChange2(BaseUnitTest):
     c = Context()
     ga_source = GaExportedNestedDataStorage(gcp_project=c.env["project"],
                                             dataset=c.env["dataset"],
-                                            table_name=c.env["table_type"],
+                                            table_type=c.env["table_type"],
                                             date_shard=c.env["date_is_active_user_added"],
                                             )