diff --git a/experiments/llmlingua2/data_collection/filter.py b/experiments/llmlingua2/data_collection/filter.py
index 5b31417..0b703aa 100644
--- a/experiments/llmlingua2/data_collection/filter.py
+++ b/experiments/llmlingua2/data_collection/filter.py
@@ -66,7 +66,7 @@
     kept["labels"],
     kept["origin"],
     kept["comp"],
-    res_pt["retrieval"],
+    kept["retrieval"],
     kept["comp_rate"],
     kept["variation_rate"],
     kept["hitting_rate"],