diff --git a/testing/test_full.py b/testing/test_full.py
index eccf0b3c8..83e6b064d 100644
--- a/testing/test_full.py
+++ b/testing/test_full.py
@@ -20,6 +20,7 @@
     get_patch_size_in_microns,
     convert_to_tiff,
 )
+from GANDLF.utils.pydantic_config import Parameters
 from GANDLF.config_manager import ConfigManager
 from GANDLF.parseConfig import parseConfig
 from GANDLF.training_manager import TrainingManager