diff --git a/downloads-generation/data_evaluation/join_with_precomputed.py b/downloads-generation/data_evaluation/join_with_precomputed.py
index bcd1b986a4faa35b31374f1c081139df9cfb1c3f..6164fbd678131e2e24cc87c5feea506ff7e5f4a5 100644
--- a/downloads-generation/data_evaluation/join_with_precomputed.py
+++ b/downloads-generation/data_evaluation/join_with_precomputed.py
@@ -118,7 +118,8 @@ def run():
                     prediction_df.index, allele
                 ]
             df.loc[sub_df.index, name] = prediction_df.max(1, skipna=False).values
-            df.loc[sub_df.index, name + " allele"] = prediction_df.idxmax(1, skipna=False).values
+            df.loc[sub_df.index, name + "_best_allele"] = prediction_df.idxmax(
+                1, skipna=False).values
 
     if 'netmhcpan4.ba' in args.predictors:
         # unflip the values