diff --git a/setup.py b/setup.py
index bf2daae37e..429f2d433d 100644
--- a/setup.py
+++ b/setup.py
@@ -5,13 +5,31 @@
 LICENSE file in the root directory of this source tree.
 """
 
-from setuptools import find_packages, setup
+from distutils.util import convert_path
+from pathlib import Path
+
+from setuptools import setup
+
+
+def make_ocpmodels_package_dict():
+    dirs = [
+        convert_path(str(p))
+        for p in Path("./ocpmodels/").glob("**")
+        if (p / "__init__.py").exists()
+    ]
+    pkgs = [d.replace("/", ".") for d in dirs]
+    return {p: d for p, d in zip(pkgs, dirs)}
+
+
+pkg_dict = make_ocpmodels_package_dict()
+pkg_dict["ocdata"] = convert_path("ocdata")
 
 setup(
-    name="ocp-models",
-    version="0.0.3",
+    name="ocpmodels",
+    version="0.0.1",
     description="Machine learning models for use in catalysis as part of the Open Catalyst Project",
     url="https://github.com/Open-Catalyst-Project/ocp",
-    packages=find_packages(),
+    packages=list(pkg_dict.keys()),
+    package_dir=pkg_dict,
     include_package_data=True,
 )