diff --git a/deepof/train_utils.py b/deepof/train_utils.py
index 2ba52061856d399ad4b9a6bc52b6b1dd30bb9587..b874c11fc9923d6a877aed318b2c3295a5401592 100644
--- a/deepof/train_utils.py
+++ b/deepof/train_utils.py
@@ -113,7 +113,7 @@ def get_callbacks(
     elif reg_cat_clusters and reg_cluster_variance:
         latreg = "categorical+variance"
 
-    run_ID = "{}{}{}{}{}{}_{}".format(
+    run_ID = "{}{}{}{}{}{}{}_{}".format(
         ("GMVAE" if variational else "AE"),
         ("_Pred={}".format(predictor) if predictor > 0 and variational else ""),
         ("_Pheno={}".format(phenotype_class) if phenotype_class > 0 else ""),