From 39f8aefa8c6e683d3d6eed2b3b098ee4231f90a8 Mon Sep 17 00:00:00 2001
From: Ben Kunkle <ben@zed.dev>
Date: Wed, 19 Nov 2025 06:44:58 -0800
Subject: [PATCH] zeta2: Improve context retrieval (#43014)

Closes #ISSUE

Release Notes:

- N/A *or* Added/Fixed/Improved ...

Co-authored-by: Agus <agus@zed.dev>
Co-authored-by: Max <max@zed.dev>
---
 crates/zeta_cli/src/evaluate.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/crates/zeta_cli/src/evaluate.rs b/crates/zeta_cli/src/evaluate.rs
index 14dc0f6c0c105919b822b9077211a1e1d9686d04..d808e3d743d7009ca66a75b3a349914b0a4f5447 100644
--- a/crates/zeta_cli/src/evaluate.rs
+++ b/crates/zeta_cli/src/evaluate.rs
@@ -478,7 +478,7 @@ fn evaluate(example: &Example, preds: &PredictionDetails, predict: bool) -> Eval
 
             let scores = Scores::new(&expected, &actual_context_lines);
 
-            false_positive_lines.retain(|line| !actual_context_lines.contains(line));
+            false_positive_lines.retain(|line| !expected.contains(line));
 
             if best_alternative_score
                 .as_ref()