diff --git a/tests/test_train_ae.py b/tests/test_train_ae.py
index 49d7e6178..6d57b7646 100644
--- a/tests/test_train_ae.py
+++ b/tests/test_train_ae.py
@@ -106,6 +106,10 @@ def test_train_ae(cmdopt, tmp_path, test_name):
             base_dir=base_dir,
             no_gpu=cmdopt["no-gpu"],
             adapt_base_dir=cmdopt["adapt-base-dir"],
+        )
+        json_data_out = modify_maps(
+            maps=json_data_out,
+            base_dir=base_dir,
             ssda=True,
         )
     assert json_data_out == json_data_ref  # ["mode"] == mode
diff --git a/tests/test_train_cnn.py b/tests/test_train_cnn.py
index 35c51ec77..9c7621985 100644
--- a/tests/test_train_cnn.py
+++ b/tests/test_train_cnn.py
@@ -127,6 +127,11 @@ def test_train_cnn(cmdopt, tmp_path, test_name):
             adapt_base_dir=cmdopt["adapt-base-dir"],
             ssda=True,
         )
+        json_data_out = modify_maps(
+            maps=json_data_out,
+            base_dir=base_dir,
+            ssda=True,
+        )
     assert json_data_out == json_data_ref  # ["mode"] == mode
 
     assert compare_folders(