diff --git a/src/re/multi_grid/grid_impl.py b/src/re/multi_grid/grid_impl.py
index b3fdfc844504bf8427fa000dcc2dae229e244474..76b9a36fe76ee1319f47d66490943eb2c15e4897 100644
--- a/src/re/multi_grid/grid_impl.py
+++ b/src/re/multi_grid/grid_impl.py
@@ -450,7 +450,8 @@ class BrokenLogGridAtLevel(SimpleOpenGridAtLevel):
         funclist = [
             lambda rg: self._gamma / (rg - self._delta),
             lambda rg: self._r_min + self._alpha * (rg - self._rg_min),
-            lambda rg: self._r_linthresh * jnp.exp(self._beta * (rg - self._rg_linthresh)),
+            lambda rg: self._r_linthresh
+            * jnp.exp(self._beta * (rg - self._rg_linthresh)),
             lambda rg: self._r_max + self._epsilon * (rg - self._rg_max),
         ]
         return jnp.piecewise(coord, condlist, funclist)
@@ -492,7 +493,7 @@ def BrokenLogGrid(
     The grid is parametrised by three radii: r_min, r_linthresh, and r_max.
     Between r_min and r_linthresh pixels are spaced linearly (r).
     Between r_linthresh and r_max pixels are spaced logarithmically (exp(r)).
-    
+
     For available parameters see the `SimpleOpenGrid` docstring in addition the ones below.
 
     Parameters
@@ -503,8 +504,9 @@ def BrokenLogGrid(
         Coordinate value at which the grid switches from linear to logarithmic spacing.
     r_max:
         Maximum coordinate value.
-    
-    Notes:
+
+    Notes
+    -----
     For values below rmin, the (padded) pixels are spaced antilinearly (1/r).
     Above rmax they are spaced linearly (r).
     """