diff --git a/FUNCTIONS.R b/FUNCTIONS.R
index 33c5bb17194a9e9b9a92905e59edeb91dba16394..8c4a358f774a003ede2c56b539ca9b428d86a5ec 100644
--- a/FUNCTIONS.R
+++ b/FUNCTIONS.R
@@ -752,7 +752,7 @@ getSeropositiveResults_RF = function(PATHWAY_1,
     RFOREST_MODEL_VOTES = rowSums(RFOREST_MODEL_PREDICTION$individual=="old")/N_tree
     
     # Merge results back to RAU
-    colnames(RAU_Dilution_Subset) = RAU_RFOREST_NAMES
+    colnames(RAU_Dilution_Subset) = RAU_RFOREST_NAMES[match(colnames(RAU_Dilution_Subset), substr(RAU_RFOREST_NAMES, 1, 3))]
   }
   
   # Select model chosen: W47
@@ -780,7 +780,7 @@ getSeropositiveResults_RF = function(PATHWAY_1,
     RFOREST_MODEL_VOTES = rowSums(RFOREST_MODEL_PREDICTION$individual=="old")/N_tree
     
     # Merge results back to RAU
-    colnames(RAU_Dilution_Subset) = RAU_RFOREST_NAMES
+    colnames(RAU_Dilution_Subset) = RAU_RFOREST_NAMES[match(colnames(RAU_Dilution_Subset), substr(RAU_RFOREST_NAMES, 1, 3))]
   }
   
   ##################################################################
@@ -1288,7 +1288,7 @@ getSeropositiveResults_SVM = function(PATHWAY_1,
     SVM_MODEL_VOTES = attr(SVM_MODEL_PREDICTION, "decision.values")[,1]
     
     # Merge results back to RAU
-    colnames(RAU_Dilution_Subset) = RAU_RFOREST_NAMES
+    colnames(RAU_Dilution_Subset) = RAU_RFOREST_NAMES[match(colnames(RAU_Dilution_Subset), substr(RAU_RFOREST_NAMES, 1, 3))]
   }
   
   # Select model chosen: W47
@@ -1316,7 +1316,7 @@ getSeropositiveResults_SVM = function(PATHWAY_1,
     SVM_MODEL_VOTES = attr(SVM_MODEL_PREDICTION, "decision.values")[,1]
     
     # Merge results back to RAU
-    colnames(RAU_Dilution_Subset) = RAU_RFOREST_NAMES
+    colnames(RAU_Dilution_Subset) = RAU_RFOREST_NAMES[match(colnames(RAU_Dilution_Subset), substr(RAU_RFOREST_NAMES, 1, 3))]
   }
   
   ##################################################################