diff --git a/tests/math/test_jcr.py b/tests/math/test_jcr.py
index fbeda53..233f1fa 100644
--- a/tests/math/test_jcr.py
+++ b/tests/math/test_jcr.py
@@ -86,7 +86,7 @@ def model(x, beta):
         return beta[0]*x**2 + beta[1]*x**5
 
     ndata = 101
-    X = np.linspace(0, 1., ndata)
+    X = np.linspace(0., 1., ndata)
     beta = (0.15, -0.05)
     Y = model(X, beta) + np.random.normal(0., 0.05, len(X))
 
@@ -94,10 +94,10 @@ def sse(beta) -> float:
         Ye = model(X, beta)
         return sum((Ye - Y)**2)
 
-    sol = minimize(sse, (-1., 1))
+    sol = minimize(sse, (-1., 1.))
     beta_est = sol.x
 
     jcr = confidence_region(beta_est,
-                            sse=lambda x, y: sse((x, y)),
+                            sse=sse,
                             ndata=ndata)
     assert len(jcr[0]) > 1