diff --git a/pyproject.toml b/pyproject.toml
index 73ec5ab8..4197ddf7 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -139,8 +139,7 @@ test-types = [
 ]
 types = [
     "sphinx-api-relink",
-    "tensorwaves[jax]",
-    "tensorwaves[pwa]",
+    "tensorwaves[jax,pwa]",
     "types-PyYAML",
     "types-requests",
     "types-tensorflow",
diff --git a/uv.lock b/uv.lock
index 627c6f76..52b98eec 100644
--- a/uv.lock
+++ b/uv.lock
@@ -4226,9 +4226,7 @@ dev = [
     { name = "sphinx-thebe" },
     { name = "sphinx-togglebutton" },
     { name = "sphobjinv" },
-    { name = "tensorwaves", extras = ["jax"] },
     { name = "tensorwaves", extras = ["jax", "pwa"] },
-    { name = "tensorwaves", extras = ["pwa"] },
     { name = "types-pyyaml" },
     { name = "types-requests" },
     { name = "types-tensorflow" },
@@ -4290,8 +4288,7 @@ style = [
     { name = "pytest" },
     { name = "pytest-mock" },
     { name = "sphinx-api-relink" },
-    { name = "tensorwaves", extras = ["jax"] },
-    { name = "tensorwaves", extras = ["pwa"] },
+    { name = "tensorwaves", extras = ["jax", "pwa"] },
     { name = "types-pyyaml" },
     { name = "types-requests" },
     { name = "types-tensorflow" },
@@ -4317,8 +4314,7 @@ types = [
     { name = "pytest" },
     { name = "pytest-mock" },
     { name = "sphinx-api-relink" },
-    { name = "tensorwaves", extras = ["jax"] },
-    { name = "tensorwaves", extras = ["pwa"] },
+    { name = "tensorwaves", extras = ["jax", "pwa"] },
     { name = "types-pyyaml" },
     { name = "types-requests" },
     { name = "types-tensorflow" },