diff --git a/model_data/simulation_system/training.py b/model_data/simulation_system/training.py
index 561d1e1d..4d751c9b 100644
--- a/model_data/simulation_system/training.py
+++ b/model_data/simulation_system/training.py
@@ -102,7 +102,7 @@ def training(
 
     if model_type == "autogluon":
         model_root = f"{target_column}-{hyperparameters['presets']}-{hyperparameters['time_limit']}-{TIMESTAMP}".lower()
-        output_base = Path(MODEL_DIRECTORY) / target_column / model_type / model_root
+        output_base = BASE_REGISTRY_PATH / target_column / model_type / model_root
 
         model = AutogluonModel(
             output_filepath=output_base / MODEL_FOLDER
@@ -164,7 +164,6 @@ def training(
     # TODO: Need a model registry - for now have this as a CSV
     # Save this in the model directory
     logger.info("--- Append registry with new model ---")
-
     registry_path = BASE_REGISTRY_PATH / target_column / REGISTRY_FILE
 
     if registry_path.exists():