diff --git a/jaxley/modules/base.py b/jaxley/modules/base.py
index b0ddc542..5ac3a4aa 100644
--- a/jaxley/modules/base.py
+++ b/jaxley/modules/base.py
@@ -27,8 +27,8 @@
     _compute_index_of_child,
     _compute_num_children,
     compute_axial_conductances,
-    compute_levels,
     compute_current_density,
+    compute_levels,
     interpolate_xyz,
     loc_of_index,
     params_to_pstate,