diff --git a/setup.py b/setup.py
index 5f86544..82fca3f 100644
--- a/setup.py
+++ b/setup.py
@@ -17,16 +17,18 @@
 with open('README.md') as readme_file:
     readme = readme_file.read()
 
+version = "0.4.0"
+
 setup(
     name="evidential_deep_learning",
-    version="0.3.0",
+    version=version,
     packages=find_packages(),
     description=
     "Learn fast, scalable, and calibrated measures of uncertainty using neural networks!",
     long_description=readme,
     long_description_content_type='text/markdown',
     url="https://github.com/aamini/evidential-deep-learning",
-    download_url="https://github.com/aamini/evidential-deep-learning/archive/v0.3.0.tar.gz",
+    download_url=f"https://github.com/aamini/evidential-deep-learning/archive/v{version}.tar.gz",
     author="Alexander Amini",
     author_email="amini@mit.edu",
     license="Apache License 2.0",