diff --git a/pyproject.toml b/pyproject.toml
index 1bd40cd0..57519439 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -21,8 +21,8 @@ maintainers = [
     "Niels Rogge <niels.rogge@ml6.eu>",
 ]
 repository = "https://github.com/ml6team/fondant"
-include = ["*.txt", "*.rst", "*.md", "fondant/components/**/*.yaml"]
-exclude = ["fondant/components"]  # Exclude everything except for component specs
+include = ["*.txt", "*.rst", "*.md", "src/fondant/components/**/*.yaml"]
+#exclude = ["fondant/components"]  # Exclude everything except for component specs
 classifiers = [
     "Development Status :: 2 - Pre-Alpha",
     "Intended Audience :: Developers",
@@ -75,6 +75,9 @@ vertex = ["docker", "kfp", "google-cloud-aiplatform"]
 sagemaker = ["sagemaker", "boto3"]
 docker = ["docker"]
 
+all = ["dask", "dask-cuda", "s3fs", "adlfs", "gcsfs", "docker", "kfp", "google-cloud-aiplatform",
+    "sagemaker", "boto3"]
+
 [tool.poetry.group.test]
 optional = true
 
diff --git a/tox.ini b/tox.ini
index 0c417a9b..eebe3a8e 100644
--- a/tox.ini
+++ b/tox.ini
@@ -43,9 +43,10 @@ allowlist_externals=
     /usr/bin/bash
 commands_pre=
     bash ./scripts/pre-build.sh
-    poetry lock
-    poetry install --all-extras
-    poetry show
+    poetry build
+    poetry run pip install dist/fondant-0.1.dev0-py3-none-any.whl
+    poetry run pip list
+    bash ./scripts/post-build.sh
 commands=
     poetry run python -m pytest tests -vv --cov fondant --cov-report term-missing --ignore=tests/integration_tests
 commands_post=