diff --git a/python/sdist/amici/jax/ode_export.py b/python/sdist/amici/jax/ode_export.py
index 95a85f6991..cb7a299a8b 100644
--- a/python/sdist/amici/jax/ode_export.py
+++ b/python/sdist/amici/jax/ode_export.py
@@ -1,12 +1,12 @@
 """
-C++ Export
+JAX Export
 ----------
-This module provides all necessary functionality to specify a differential
-equation model and generate executable C++ simulation code.
+This module provides all necessary functionality to specify an ordinary
+differential equation model and generate executable jax simulation code.
 The user generally won't have to directly call any function from this module
 as this will be done by
-:py:func:`amici.pysb_import.pysb2amici`,
-:py:func:`amici.sbml_import.SbmlImporter.sbml2amici` and
+:py:func:`amici.pysb_import.pysb2jax`,
+:py:func:`amici.sbml_import.SbmlImporter.sbml2jax` and
 :py:func:`amici.petab_import.import_model`.
 """