diff --git a/tests/test_commands.py b/tests/test_commands.py
index 9d62c4ddda1df85fc9c170844cef8c9c5a50a4f1..4192bb5b6753ec7949c268a86b1b17f25b02f624 100644
--- a/tests/test_commands.py
+++ b/tests/test_commands.py
@@ -4,7 +4,7 @@ import pkg_resources
 
 from click.testing import CliRunner
 
-from bob.extension.scripts.click_helper import assert_click_runner_result
+from bob.io.base.testing_utils import assert_click_runner_result
 from bob.pad.base.script import pad_commands
 
 
diff --git a/tests/test_pipelines.py b/tests/test_pipelines.py
index b6bf2b2768fb49bcf6e748134404d911fbe1fa24..39c8efdabc64a5da1cd3ea7556be8925c505322c 100644
--- a/tests/test_pipelines.py
+++ b/tests/test_pipelines.py
@@ -11,7 +11,7 @@ from sklearn.pipeline import Pipeline
 import bob.measure
 import bob.pipelines as mario
 
-from bob.extension.scripts.click_helper import assert_click_runner_result
+from bob.io.base.testing_utils import assert_click_runner_result
 from bob.pad.base.error_utils import split_csv_pad
 from bob.pad.base.pipelines import Database
 from bob.pad.base.script.run_pipeline import run_pipeline as run_pipeline_cli