diff --git a/tests/test_cli.py b/tests/test_cli.py
index 3d6750326f6653eb4c9cfe539d5c5ee0854a2643..888794b4d1fe5cae20f01f524de726d090224e41 100644
--- a/tests/test_cli.py
+++ b/tests/test_cli.py
@@ -513,17 +513,17 @@ def test_compare_pasa_montgomery(temporary_basedir):
 
 @pytest.mark.skip(reason="Test need to be updated")
 @pytest.mark.skip_if_rc_var_not_set("datadir.montgomery")
-def test_train_signstotb_montgomery_rs(temporary_basedir, datadir):
+def test_train_mlp_montgomery_rs(temporary_basedir, datadir):
     from ptbench.scripts.train import train
 
     runner = CliRunner()
 
     with stdout_logging() as buf:
-        output_folder = str(temporary_basedir / "results/signstotb")
+        output_folder = str(temporary_basedir / "results/mlp")
         result = runner.invoke(
             train,
             [
-                "signs_to_tb",
+                "mlp",
                 "montgomery_rs",
                 "-vv",
                 "--epochs=1",
@@ -567,7 +567,7 @@ def test_train_signstotb_montgomery_rs(temporary_basedir, datadir):
 
 @pytest.mark.skip(reason="Test need to be updated")
 @pytest.mark.skip_if_rc_var_not_set("datadir.montgomery")
-def test_predict_signstotb_montgomery_rs(temporary_basedir, datadir):
+def test_predict_mlp_montgomery_rs(temporary_basedir, datadir):
     from ptbench.scripts.predict import predict
 
     runner = CliRunner()
@@ -577,12 +577,12 @@ def test_predict_signstotb_montgomery_rs(temporary_basedir, datadir):
         result = runner.invoke(
             predict,
             [
-                "signs_to_tb",
+                "mlp",
                 "montgomery_rs",
                 "-vv",
                 "--batch-size=1",
                 "--relevance-analysis",
-                f"--weight={str(datadir / 'lfs' / 'models' / 'signstotb.ckpt')}",
+                f"--weight={str(datadir / 'lfs' / 'models' / 'mlp.ckpt')}",
                 f"--output-folder={output_folder}",
             ],
         )