diff --git a/setup.py b/setup.py
index 872befa..1c9e876 100644
--- a/setup.py
+++ b/setup.py
@@ -2,7 +2,7 @@
 
 setup(
     name="optillm",
-    version="0.0.2",
+    version="0.0.4",
     packages=find_packages(),
     py_modules=['optillm'],
     install_requires=[