diff --git a/docs/conf.py b/docs/conf.py
index 48e9a28..f1bbd1d 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -11,7 +11,7 @@
 from plotly.io._sg_scraper import plotly_sg_scraper
 
 # Path to the VERSION file
-version_file = pathlib.Path(__file__).parent.parent / "VERSION"
+version_file = pathlib.Path(__file__).parent.parent / "mloptimizer" / "VERSION"
 version = version_file.read_text().strip()
 
 sys.path.insert(0, os.path.abspath('..'))
diff --git a/VERSION b/mloptimizer/VERSION
similarity index 100%
rename from VERSION
rename to mloptimizer/VERSION
diff --git a/mloptimizer/__main__.py b/mloptimizer/__main__.py
index ad5ba30..6ed4fb6 100644
--- a/mloptimizer/__main__.py
+++ b/mloptimizer/__main__.py
@@ -3,7 +3,7 @@
 
 def get_version():
     # Read the version from the VERSION file located one directory up
-    version_file = Path(__file__).resolve().parent.parent / "VERSION"
+    version_file = Path(__file__).resolve().parent / "VERSION"
     with open(version_file) as f:
         return f.read().strip()
 
diff --git a/setup.py b/setup.py
index dfd9d86..238360c 100644
--- a/setup.py
+++ b/setup.py
@@ -4,7 +4,7 @@
 
 here = pathlib.Path(__file__).parent.resolve()
 long_description = (here / "README.md").read_text(encoding="utf-8")
-version = (here / "VERSION").read_text().strip()
+version = (here / "mloptimizer" / "VERSION").read_text().strip()
 
 
 
@@ -140,7 +140,7 @@ def read_requirements(requirements_file):
     # If there are data files included in your packages that need to be
     # installed, specify them here.
     package_data={  # Optional
-        "mloptimizer": ["infrastructure/config/**/*"],
+        "mloptimizer": ["infrastructure/config/**/*", "VERSION"],
     },
     # Entry points. The following would provide a command called `sample` which
     # executes the function `main` from this package when invoked: