diff --git a/pypunisher/__init__.py b/pypunisher/__init__.py
index 4952b6d..4dd32a4 100644
--- a/pypunisher/__init__.py
+++ b/pypunisher/__init__.py
@@ -13,5 +13,5 @@
 from pypunisher.metrics.criterion import aic, bic
 from pypunisher.selection_engines.selection import Selection
 
-__version_info__ = (4, 0, 0)
+__version_info__ = (4, 0, 1)
 __version__ = '.'.join(map(str, __version_info__))
diff --git a/setup.py b/setup.py
index 1c34f6f..00b1afa 100644
--- a/setup.py
+++ b/setup.py
@@ -11,7 +11,7 @@ def read(fname):
 
 setup(
     name='pypunisher',
-    version='4.0.0',
+    version='4.0.1',
     author='Jill Cates, Avinash Prabhakaran, Tariq Hassan',
     author_email='NA',
     description='Model Selection in Python',
@@ -19,7 +19,7 @@ def read(fname):
     license='BSD-3',
     keywords='model selection',
     url='https://github.com/UBC-MDS/PyPunisher',
-    download_url='https://github.com/UBC-MDS/PyPunisher/archive/v4.0.0tar.gz',
+    download_url='https://github.com/UBC-MDS/PyPunisher/archive/v4.0.1tar.gz',
     packages=find_packages(exclude=("tests",)),
     # Note: requirements.txt contains some packages
     # which are not needed to simply use the package