diff --git a/test/test_run_modules/test_training.py b/test/test_run_modules/test_training.py
index d6a7f8297f4dfb674bd4109ab0a62f6313990590..c0b625ef70deeb0686b236275e6bd1182ad48d41 100644
--- a/test/test_run_modules/test_training.py
+++ b/test/test_run_modules/test_training.py
@@ -187,7 +187,7 @@ class TestTraining:
         obj.data_store.set("hist", hist, "general.model")
         obj.data_store.set("experiment_name", "TestExperiment", "general")
         obj.data_store.set("experiment_path", path, "general")
-        obj.data_store.set("trainable", True, "general")
+        obj.data_store.set("train_model", True, "general")
         obj.data_store.set("create_new_model", True, "general")
         os.makedirs(batch_path)
         obj.data_store.set("batch_path", batch_path, "general")