diff --git a/expo/experimenter/experimenter.py b/expo/experimenter/experimenter.py
index b1b5a93c0..418e0089a 100644
--- a/expo/experimenter/experimenter.py
+++ b/expo/experimenter/experimenter.py
@@ -85,7 +85,8 @@ class Experimenter:
         os.makedirs(state["node_dir"], exist_ok=True)
         pred_node_path = os.path.join(state["node_dir"], f"{self.start_time}-{split}_predictions.csv")
         gt_path = os.path.join(state["datasets_dir"][f"{split}_target"])
-        preds = pd.read_csv(pred_path)["target"]
+        preds = pd.read_csv(pred_path)
+        preds = preds[preds.columns.tolist()[0]]
         preds.to_csv(pred_node_path, index=False)
         gt = pd.read_csv(gt_path)["target"]
         metric = state["dataset_config"]["metric"]