diff --git a/how_to_upload_to_pypi b/how_to_upload_to_pypi
index f3cb17f..aff537f 100644
--- a/how_to_upload_to_pypi
+++ b/how_to_upload_to_pypi
@@ -13,3 +13,4 @@ python -m twine upload dist/*
 
 # test
 pip install autodiscjax
+pip uninstall autodiscjax
diff --git a/setup.py b/setup.py
index 82c7d9e..11a94c8 100644
--- a/setup.py
+++ b/setup.py
@@ -6,7 +6,7 @@
 
 setuptools.setup(
     name='autodiscjax',
-    version='0.4',
+    version='0.4.1',
     author='Mayalen Etcheverry',
     author_email='mayalen.etcheverry@inria.fr',
     description=' python library built on top of jax to facilitate automated exploration and simulation of computational models of biological processes',