diff --git a/Load.py b/Load.py
index 32059d2695e75e6d401afa99c0e17f2c487442a7..b197b3b911bd7d64af540046b102eedd63424d71 100644
--- a/Load.py
+++ b/Load.py
@@ -479,6 +479,6 @@ def concatenateRuns(runs):
             return
     
     result = xr.concat(orderedRuns, dim='trainId')
-    result.attrs['run'] = [run.attrs['run'] for run in orderedRuns]
-    result.attrs['runFolder'] = [run.attrs['runFolder'] for run in orderedRuns]
+    for k in orderedRuns[0].attrs.keys():
+        result.attrs[k] = [run.attrs[k] for run in orderedRuns]
     return result