diff --git a/docs/conf.py b/docs/conf.py
index 2d933c7..b2cee40 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -5,7 +5,7 @@
 autodoc_member_order = "groupwise"
 exclude_patterns = ["_build", "**tests**"]
 
-from jaxparrow import __version__
+from version import __version__
 
 
 # Configuration file for the Sphinx documentation builder.
diff --git a/jaxparrow/__init__.py b/jaxparrow/__init__.py
index e787a58..53c6934 100644
--- a/jaxparrow/__init__.py
+++ b/jaxparrow/__init__.py
@@ -1,5 +1,5 @@
+from .version import __version__
 from .cyclogeostrophy import cyclogeostrophy
 from .geostrophy import geostrophy
 
-__all__ = ["cyclogeostrophy", "geostrophy"]
-__version__ = "0.0.1"
+__all__ = ["__version__", "cyclogeostrophy", "geostrophy"]
diff --git a/jaxparrow/version.py b/jaxparrow/version.py
new file mode 100644
index 0000000..f102a9c
--- /dev/null
+++ b/jaxparrow/version.py
@@ -0,0 +1 @@
+__version__ = "0.0.1"
diff --git a/notebooks/README.md b/notebooks/README.md
index 841c9bc..d44f99d 100644
--- a/notebooks/README.md
+++ b/notebooks/README.md
@@ -1,5 +1,3 @@
-# Notebooks
-
 We designed two notebooks, one with the extremely idealised scenario of a [gaussian eddy](gaussian_eddy), and a realistic one focusing on the [Alboran sea](alboran_sea) area.
 
 The goal is both to showcase step-by-step how **jaxparrow** can be used, and to demonstrate the interest of the variational approach we propose.
\ No newline at end of file
diff --git a/pyproject.toml b/pyproject.toml
index 29e6b7f..f9662a9 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -42,7 +42,7 @@ jaxparrow = "jaxparrow:main"
 "Bug Tracker" = "https://github.com/meom-group/jaxparrow/issues"
 
 [tool.setuptools.dynamic]
-version = {attr = "jaxparrow.__version__"}
+version = {attr = "jaxparrow.version.__version__"}
 
 [tool.setuptools.packages]
 find = {}