diff --git a/setup.py b/setup.py
index 28485de..5355c03 100644
--- a/setup.py
+++ b/setup.py
@@ -8,14 +8,14 @@
 
 setuptools.setup(
     name="neurodiffeq",
-    version="0.3.3",
+    version="0.3.4",
     author="neurodiffgym",
     author_email="shuheng_liu@g.harvard.edu",
     description="A light-weight & flexible library for solving differential equations using neural networks based on PyTorch. ",
     long_description=long_description,
     long_description_content_type="text/markdown",
     url="https://github.com/NeuroDiffGym/neurodiffeq",
-    download_url="https://github.com/NeuroDiffGym/neurodiffeq/archive/v0.3.3.tar.gz",
+    download_url="https://github.com/NeuroDiffGym/neurodiffeq/archive/v0.3.4.tar.gz",
     keywords=[
         "neural network",
         "deep learning",