1 files changed, 1 insertions, 1 deletions
diff --git a/utils/chi2 b/utils/chi2
index bb62af3..f600133 100755
--- a/utils/chi2
+++ b/utils/chi2
@@ -236,7 +236,7 @@ def make_nll(data, muons, mc, bins, print_nll=False):
                                          data.loc[data.id == id,'energy2'].values]).T
 
     def nll(x, grad=None):
-        if any(x[i] < 0 for i in range(len(x))):
+        if any(x[i] < 0 for i in (0,2,4,5,6)):
             return np.inf
 
         # Get the data histograms. We need to do this within the