diff --git a/python/sdist/amici/jax/model.py b/python/sdist/amici/jax/model.py
index 126cdb8039..cecebeab0e 100644
--- a/python/sdist/amici/jax/model.py
+++ b/python/sdist/amici/jax/model.py
@@ -196,6 +196,7 @@ def _nllh(
     def state_ids(self) -> list[str]:
         """
         Get the state ids of the model.
+
         :return:
             State ids
         """
@@ -206,6 +207,7 @@ def state_ids(self) -> list[str]:
     def observable_ids(self) -> list[str]:
         """
         Get the observable ids of the model.
+
         :return:
             Observable ids
         """
@@ -216,6 +218,7 @@ def observable_ids(self) -> list[str]:
     def parameter_ids(self) -> list[str]:
         """
         Get the parameter ids of the model.
+
         :return:
             Parameter ids
         """