diff --git a/tensorflow_datasets/testing/__init__.py b/tensorflow_datasets/testing/__init__.py
index 3a941737635..e13788302fb 100644
--- a/tensorflow_datasets/testing/__init__.py
+++ b/tensorflow_datasets/testing/__init__.py
@@ -42,6 +42,7 @@
   from tensorflow_datasets.testing.test_utils import dummy_croissant_file
   from tensorflow_datasets.testing.test_utils import DummyBeamDataset
   from tensorflow_datasets.testing.test_utils import DummyDataset
+  from tensorflow_datasets.testing.test_utils import DummyDatasetWithConfigBuilder
   from tensorflow_datasets.testing.test_utils import DummyDatasetCollection
   from tensorflow_datasets.testing.test_utils import DummyDatasetSharedGenerator
   from tensorflow_datasets.testing.test_utils import DummyMnist
@@ -70,6 +71,7 @@
     "dummy_croissant_file": "tensorflow_datasets.testing.test_utils",
     "DummyBeamDataset": "tensorflow_datasets.testing.test_utils",
     "DummyDataset": "tensorflow_datasets.testing.test_utils",
+    "DummyDatasetWithConfigBuilder": "tensorflow_datasets.testing.test_utils",
     "DummyDatasetCollection": "tensorflow_datasets.testing.test_utils",
     "DummyDatasetSharedGenerator": "tensorflow_datasets.testing.test_utils",
     "DummyMnist": "tensorflow_datasets.testing.test_utils",
diff --git a/tensorflow_datasets/testing/test_utils.py b/tensorflow_datasets/testing/test_utils.py
index eccc8d9f1bf..df9ab1de152 100644
--- a/tensorflow_datasets/testing/test_utils.py
+++ b/tensorflow_datasets/testing/test_utils.py
@@ -38,13 +38,14 @@
 from tensorflow_datasets.core import dataset_info
 from tensorflow_datasets.core import example_serializer
 from tensorflow_datasets.core import features
+from tensorflow_datasets.core import features as features_lib
 from tensorflow_datasets.core import lazy_imports_lib
 from tensorflow_datasets.core import naming
+from tensorflow_datasets.core import splits as splits_lib
 from tensorflow_datasets.core import utils
 from tensorflow_datasets.core.utils.lazy_imports_utils import mlcroissant as mlc
 from tensorflow_datasets.core.utils.lazy_imports_utils import tensorflow as tf
 
-
 _GCS_ACCESS_FNS = {
     'original_info': utils.gcs_utils.gcs_dataset_info_files,
     'dummy_info': lambda _: [],