diff --git a/tests/pipelines/test_vanilla_pipeline.py b/tests/pipelines/test_vanilla_pipeline.py
index 2218294b7..b591000ab 100644
--- a/tests/pipelines/test_vanilla_pipeline.py
+++ b/tests/pipelines/test_vanilla_pipeline.py
@@ -11,10 +11,10 @@ from nerfstudio.data.datasets.base_dataset import DataparserOutputs, InputDatase
 from nerfstudio.pipelines.base_pipeline import (
     Model,
     ModelConfig,
-    VanillaDataManagerConfig,
     VanillaPipeline,
     VanillaPipelineConfig,
 )
+from nerfstudio.data.base_datamanager import VanillaDataManagerConfig
 
 
 class MockedDataManager: