1 files changed, 1 insertions, 1 deletions
diff --git a/utils/chi2 b/utils/chi2
index 7f24c1f..67ed743 100755
--- a/utils/chi2
+++ b/utils/chi2
@@ -172,7 +172,7 @@ def make_nll(data, muons, mc, bins):
 
     ke_dict = {}
     for id in (20,22,2020,2022,2222):
-        ke_dict[id] = data[data.id == id].ke.values
+        ke_dict[id] = mc[mc.id == id].ke.values
 
     def nll(x, grad=None):
         if any(x[i] < 0 for i in range(len(x))):