diff --git a/src/agentlab/analyze/agent_xray.py b/src/agentlab/analyze/agent_xray.py
index 0d643600..8accbfd6 100644
--- a/src/agentlab/analyze/agent_xray.py
+++ b/src/agentlab/analyze/agent_xray.py
@@ -103,7 +103,16 @@ def update_exp_result(self, episode_id: EpisodeId):
 
         # find unique row using idx
         result_df = self.agent_df.reset_index(inplace=False)
-        exp_dir = result_df.iloc[episode_id.row_index]["exp_dir"]
+        sub_df = result_df[result_df["_row_index"] == episode_id.row_index]
+        if len(sub_df) == 0:
+            self.exp_result = None
+            raise ValueError(f"Could not find _row_index: {episode_id.row_index}")
+
+        if len(sub_df) > 1:
+            warning(
+                f"Found multiple rows with same row_index {episode_id.row_index} Using the first one."
+            )
+        exp_dir = sub_df.iloc[0]["exp_dir"]
         print(exp_dir)
         self.exp_result = ExpResult(exp_dir)
         self.step = 0