From 2ace1f60d824852e3ccecfeafd40fb6faded72b3 Mon Sep 17 00:00:00 2001
From: Rahul <rgaur@terpmail.umd.edu>
Date: Sat, 28 Sep 2024 13:37:35 -0400
Subject: [PATCH] fixing more tests 2

---
 desc/compute/_profiles.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/desc/compute/_profiles.py b/desc/compute/_profiles.py
index 31f013ec2..17d8c5711 100644
--- a/desc/compute/_profiles.py
+++ b/desc/compute/_profiles.py
@@ -9,7 +9,7 @@
 expensive computations.
 """
 
-from quadax import simpson
+from quadax import cumulative_simpson
 from scipy.constants import elementary_charge, mu_0
 
 from desc.backend import cond, jnp
@@ -143,7 +143,7 @@ def _chi_r(params, transforms, profiles, data, **kwargs):
     resolution_requirement="r",
 )
 def _chi(params, transforms, profiles, data, **kwargs):
-    chi = simpson(
+    chi = cumulative_simpson(
         y=transforms["grid"].compress(data["chi_r"]),
         x=transforms["grid"].compress(data["rho"]),
         initial=0,