diff --git a/reagent/test/workflow/test_preprocessing.py b/reagent/test/workflow/test_preprocessing.py
index fdcaab95..f04d920a 100644
--- a/reagent/test/workflow/test_preprocessing.py
+++ b/reagent/test/workflow/test_preprocessing.py
@@ -5,6 +5,7 @@ import logging
 import unittest
 
 import numpy as np
+import pvc
 import pytest
 from reagent.preprocessing.identify_types import CONTINUOUS
 
@@ -48,7 +49,7 @@ class TestPreprocessing(ReagentSQLTestBase):
         num_samples = NUM_ROWS // 2
         preprocessing_options = PreprocessingOptions(num_samples=num_samples)
 
-        table_spec = TableSpec(table_name=TABLE_NAME)
+        table_spec = TableSpec(dataset=pvc.HiveDataset(table=TABLE_NAME))
 
         normalization_params = identify_normalization_parameters(
             table_spec, COL_NAME, preprocessing_options, seed=self.test_class_seed
diff --git a/reagent/test/workflow/test_query_data.py b/reagent/test/workflow/test_query_data.py
index 2ac6ee09..c68f7522 100644
--- a/reagent/test/workflow/test_query_data.py
+++ b/reagent/test/workflow/test_query_data.py
@@ -5,6 +5,7 @@ import logging
 import unittest
 
 import numpy as np
+import pvc
 import pytest
 
 # pyre-fixme[21]: Could not find `pyspark`.
@@ -49,7 +50,7 @@ class TestQueryData(ReagentSQLTestBase):
     def _discrete_read_data(
         self, custom_reward_expression=None, gamma=None, multi_steps=None
     ):
-        ts = TableSpec(table_name=self.table_name)
+        ts = TableSpec(dataset=pvc.HiveDataset(table=self.table_name))
         dataset: Dataset = query_data(
             input_table_spec=ts,
             discrete_action=True,
diff --git a/reagent/test/workflow/test_query_data_parametric.py b/reagent/test/workflow/test_query_data_parametric.py
index 536bfd77..d43f200d 100644
--- a/reagent/test/workflow/test_query_data_parametric.py
+++ b/reagent/test/workflow/test_query_data_parametric.py
@@ -5,6 +5,7 @@ import logging
 import unittest
 
 import numpy as np
+import pvc
 import pytest
 
 # pyre-fixme[21]: Could not find `pyspark`.
@@ -49,7 +50,7 @@ class TestQueryDataParametric(ReagentSQLTestBase):
     def _parametric_read_data(
         self, custom_reward_expression=None, gamma=None, multi_steps=None
     ):
-        ts = TableSpec(table_name=self.table_name)
+        ts = TableSpec(dataset=pvc.HiveDataset(table=self.table_name))
         dataset: Dataset = query_data(
             input_table_spec=ts,
             discrete_action=False,