diff --git a/setup.py b/setup.py
index 3954d701..88bbdbb9 100644
--- a/setup.py
+++ b/setup.py
@@ -16,7 +16,7 @@
         install_requires.append(line)
 
 setup(name="weaver-core",
-      version='0.4.0',
+      version='0.4.1',
       description="A streamlined deep-learning framework for high energy physics",
       long_description_content_type="text/markdown",
       author="H. Qu, C. Li",