diff --git a/pyproject.toml b/pyproject.toml
index 07a75e75..0e7f6631 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -64,7 +64,7 @@ alchemical-model = [
   "torch_alchemical @ git+https://github.com/abmazitov/torch_alchemical.git@51ff519",
 ]
 pet = [
-  "pet @ git+https://github.com/lab-cosmo/pet@8841021",
+  "pet @ git+https://github.com/lab-cosmo/pet@c442f32",
 ]
 gap = [
     "rascaline-torch @ git+https://github.com/luthaf/rascaline@d181b28#subdirectory=python/rascaline-torch",
diff --git a/tox.ini b/tox.ini
index a38fe494..88f72d58 100644
--- a/tox.ini
+++ b/tox.ini
@@ -116,7 +116,7 @@ extras = pet
 changedir = src/metatrain/experimental/pet/tests/
 commands =
     python -m pip uninstall -y pet
-    python -m pip install "pet @ git+https://github.com/lab-cosmo/pet@8841021" --no-build-isolation
+    python -m pip install "pet @ git+https://github.com/lab-cosmo/pet@c442f32" --no-build-isolation
     pytest {posargs}
 
 [testenv:gap-tests]