1 files changed, 4 insertions, 4 deletions
diff --git a/utils/chi2 b/utils/chi2
index 0326e1a..39db081 100755
--- a/utils/chi2
+++ b/utils/chi2
@@ -480,10 +480,10 @@ if __name__ == '__main__':
     ev_mc = ev_mc[~np.isnan(ev_mc.fmin)]
     muon_mc = muon_mc[~np.isnan(muon_mc.fmin)]
 
-    # require (r/r_psup)^3 < 0.9
-    ev = ev[ev.r_psup < 0.9]
-    ev_mc = ev_mc[ev_mc.r_psup < 0.9]
-    muon_mc = muon_mc[muon_mc.r_psup < 0.9]
+    # require (r < av radius)
+    ev = ev[ev.r < AV_RADIUS]
+    ev_mc = ev_mc[ev_mc.r < AV_RADIUS]
+    muon_mc = muon_mc[muon_mc.r < AV_RADIUS]
 
     # require psi < 6
     ev = ev[ev.psi < 6]