diff --git a/.github/workflows/ci-cd.yml b/.github/workflows/ci-cd.yml
index 85d8d9a3b..26855e175 100644
--- a/.github/workflows/ci-cd.yml
+++ b/.github/workflows/ci-cd.yml
@@ -202,10 +202,13 @@ jobs:
 
       - name: Build type stubs
         run: inv type-stubs
+      
+      - name: Install build
+        run: pip install build
 
       - name: Build package distributions (wheel and source)
         run: |
-          python setup.py sdist bdist_wheel
+          python -m build
 
       - name: Upload to PyPI
         env:
diff --git a/pyproject.toml b/pyproject.toml
index eb554feda..86c34f1fd 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -47,7 +47,7 @@ Documentation = "https://modal.com/docs"
 modal = "modal.__main__:main"
 
 [tool.setuptools.packages.find]
-include = ["modal", "modal_version", "modal_proto"]
+include = ["modal", "modal_version", "modal_proto", "modal.*"]
 exclude = ["test*", "modal_docs", "modal_global_objects"]
 
 [tool.setuptools.package-data]