diff --git a/optillm/__init__.py b/optillm/__init__.py
index 8e114d7d..61648ec9 100644
--- a/optillm/__init__.py
+++ b/optillm/__init__.py
@@ -2,7 +2,7 @@
 import os
 
 # Version information
-__version__ = "0.1.3"
+__version__ = "0.1.4"
 
 # Get the path to the root optillm.py
 spec = util.spec_from_file_location(
diff --git a/setup.py b/setup.py
index 28b0414a..8badb50c 100644
--- a/setup.py
+++ b/setup.py
@@ -1,18 +1,10 @@
+import os
 from setuptools import setup, find_packages
 
-def read_version():
-    version_dict = {}
-    try:
-        with open('optillm/__init__.py', 'r') as f:
-            exec(f.read(), version_dict)
-        return version_dict.get('__version__', 'unknown')
-    except Exception:
-        return 'unknown'
-
 setup(
     name="optillm",
-    version="0.1.2",
-    packages=find_packages(),
+    version="0.1.4",
+    packages=find_packages(include=['optillm', 'optillm.*']),  # This ensures all subpackages are included
     py_modules=['optillm'],
     package_data={
         'optillm': [