diff --git a/setup.py b/setup.py
index d3c8af2..47ffcdb 100644
--- a/setup.py
+++ b/setup.py
@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="minillmflow",
-    version="0.0.3",
+    version="0.0.4",
     packages=find_packages(),
     author="Zachary Huang",
     author_email="zh2408@columbia.edu",