diff --git a/docs/how_to_guide/plot_02_glm_demo.py b/docs/how_to_guide/plot_02_glm_demo.py
index 23ef1396..f1c6e3b2 100644
--- a/docs/how_to_guide/plot_02_glm_demo.py
+++ b/docs/how_to_guide/plot_02_glm_demo.py
@@ -25,12 +25,12 @@
 
 """
 
+import jax
 import matplotlib.pyplot as plt
 import numpy as np
 from matplotlib.patches import Rectangle
 from sklearn import model_selection
 
-import jax
 import nemos as nmo
 
 np.random.seed(111)
diff --git a/docs/how_to_guide/plot_03_glm_pytree.py b/docs/how_to_guide/plot_03_glm_pytree.py
index 22e0f5f2..2d36db3b 100644
--- a/docs/how_to_guide/plot_03_glm_pytree.py
+++ b/docs/how_to_guide/plot_03_glm_pytree.py
@@ -12,10 +12,10 @@
 First, however, let's briefly discuss FeaturePytrees.
 
 """
-import numpy as np
-
 import jax
 import jax.numpy as jnp
+import numpy as np
+
 import nemos as nmo
 
 np.random.seed(111)
diff --git a/docs/how_to_guide/plot_04_population_glm.py b/docs/how_to_guide/plot_04_population_glm.py
index a4e0eec4..70dac9cd 100644
--- a/docs/how_to_guide/plot_04_population_glm.py
+++ b/docs/how_to_guide/plot_04_population_glm.py
@@ -22,10 +22,10 @@
 Let's generate some synthetic data and fit a population model.
 """
 
+import jax.numpy as jnp
 import matplotlib.pyplot as plt
 import numpy as np
 
-import jax.numpy as jnp
 import nemos as nmo
 
 np.random.seed(123)
diff --git a/docs/tutorials/plot_01_current_injection.py b/docs/tutorials/plot_01_current_injection.py
index aeafe377..8eb83dbe 100644
--- a/docs/tutorials/plot_01_current_injection.py
+++ b/docs/tutorials/plot_01_current_injection.py
@@ -48,12 +48,12 @@
 
 
 
+# Import everything
+import jax
 import matplotlib.pyplot as plt
 import numpy as np
 import pynapple as nap
 
-# Import everything
-import jax
 import nemos as nmo
 
 # some helper plotting functions
diff --git a/docs/tutorials/plot_06_calcium_imaging.py b/docs/tutorials/plot_06_calcium_imaging.py
index 514a0367..985a0b61 100644
--- a/docs/tutorials/plot_06_calcium_imaging.py
+++ b/docs/tutorials/plot_06_calcium_imaging.py
@@ -10,12 +10,12 @@
 
 """
 
+import jax
+import jax.numpy as jnp
 import matplotlib.pyplot as plt
 import pynapple as nap
 from sklearn.linear_model import LinearRegression
 
-import jax
-import jax.numpy as jnp
 import nemos as nmo
 
 # %%
diff --git a/src/nemos/__init__.py b/src/nemos/__init__.py
index aedd05c0..97c5b3db 100644
--- a/src/nemos/__init__.py
+++ b/src/nemos/__init__.py
@@ -14,5 +14,5 @@
     styles,
     tree_utils,
     type_casting,
-    utils
+    utils,
 )
diff --git a/src/nemos/_documentation_utils/__init__.py b/src/nemos/_documentation_utils/__init__.py
index 1c64a43a..3cd63e0e 100644
--- a/src/nemos/_documentation_utils/__init__.py
+++ b/src/nemos/_documentation_utils/__init__.py
@@ -19,5 +19,5 @@
     plot_rates_and_smoothed_counts,
     plot_weighted_sum_basis,
     run_animation,
-    tuning_curve_plot
+    tuning_curve_plot,
 )