diff --git a/tests/test_lib3sgm.py b/tests/test_lib3sgm.py
index c7b526c..308cede 100644
--- a/tests/test_lib3sgm.py
+++ b/tests/test_lib3sgm.py
@@ -29,9 +29,8 @@
 import pandora
 import rasterio
 import xarray as xr
-from pandora import optimization, check_conf
+from pandora import optimization, check_conf, check_configuration
 from pandora.state_machine import PandoraMachine
-import pandora.check_json as JSON_checker
 from tests import common
 
 # pylint: disable=too-many-lines, too-many-public-methods, redefined-outer-name
@@ -1249,7 +1248,7 @@ def test_user_initiate_3sgm_and_validation_with_one_geomprior_segmentation(self,
 
         # check the configuration
         with pytest.raises(SystemExit):
-            _ = JSON_checker.check_conf(cfg, pandora_machine)
+            _ = check_configuration.check_conf(cfg, pandora_machine)
 
     def test_user_initiate_3sgm_and_validation_with_one_geomprior_classification(self, user_cfg, inputs_path):
         """
@@ -1280,4 +1279,4 @@ def test_user_initiate_3sgm_and_validation_with_one_geomprior_classification(sel
 
         # check the configuration
         with pytest.raises(SystemExit):
-            _ = JSON_checker.check_conf(cfg, pandora_machine)
+            _ = check_configuration.check_conf(cfg, pandora_machine)