diff --git a/tests/__init__.py b/tests/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/tests/test_loader.py b/tests/test_loader.py
index 5eaa9cb..7f940cc 100644
--- a/tests/test_loader.py
+++ b/tests/test_loader.py
@@ -19,14 +19,14 @@
 from absl.testing import absltest
 
 from datum.reader.loader import load
-from tests import utils
+from tests.utils import _test_create_det_records
 
 
 class TestDataset(absltest.TestCase):
 
   def setUp(self):
     self.tempdir = tempfile.mkdtemp()
-    utils._test_create_det_records(self.tempdir)
+    _test_create_det_records(self.tempdir)
     self._dataset = load(self.tempdir)
     dataset_configs = self._dataset.dataset_configs
     dataset_configs.batch_size_train = 1