diff --git a/pdm.lock b/pdm.lock
index 3d759b09..c328a71c 100644
--- a/pdm.lock
+++ b/pdm.lock
@@ -5,7 +5,7 @@
 groups = ["default", "dev", "gpu"]
 strategy = ["cross_platform", "inherit_metadata"]
 lock_version = "4.4.1"
-content_hash = "sha256:f08da5dfdfe44b3606f4b96358e55bcd3cf6794edec048ea254dbbc227260ec6"
+content_hash = "sha256:7cdab2243273dabb91eab0318c8f738607c5ff1b726c7da07a7c88863341600e"
 
 [[package]]
 name = "absl-py"
@@ -3292,11 +3292,8 @@ files = [
 
 [[package]]
 name = "tensor-regression"
-version = "0.0.2.post3.dev0"
+version = "0.0.6"
 requires_python = "<4.0,>=3.10"
-git = "https://www.github.com/lebrice/tensor_regression"
-ref = "add-skip-creation"
-revision = "cad4002a45c154f0c0cbace7aadadce4ad7a0b99"
 summary = "A small wrapper around pytest_regressions for Tensors"
 groups = ["dev"]
 dependencies = [
@@ -3304,6 +3301,10 @@ dependencies = [
     "pytest-regressions<3.0.0,>=2.5.0",
     "torch<3.0.0,>=2.0.0",
 ]
+files = [
+    {file = "tensor_regression-0.0.6-py3-none-any.whl", hash = "sha256:af1082618deda725ad40448deea672fb220bd406e77118fd69b15947eccae577"},
+    {file = "tensor_regression-0.0.6.tar.gz", hash = "sha256:b399e71b10f8c7db78e8e1a50b1c92a1aef62c852d1fe54e4c1cbcc95017aaa5"},
+]
 
 [[package]]
 name = "tensorboard"
diff --git a/project/main_test.py b/project/main_test.py
index e73dd929..fdff471b 100644
--- a/project/main_test.py
+++ b/project/main_test.py
@@ -30,7 +30,7 @@ def test_torch_can_use_the_GPU():
 @use_overrides([""])
 def test_defaults(experiment_config: Config) -> None:
     assert experiment_config.algorithm["_target_"] == (
-        ExampleAlgorithm.__module__ + "." + ExampleAlgorithm.__name__
+        ExampleAlgorithm.__module__ + "." + ExampleAlgorithm.__qualname__
     )
     assert (
         isinstance(experiment_config.datamodule, CIFAR10DataModule)
diff --git a/pyproject.toml b/pyproject.toml
index b81b9761..ee6cfc67 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -27,7 +27,6 @@ dependencies = [
     "tensorboard>=2.16.2",
     "gymnax>=0.0.8",
     "torch-jax-interop>=0.0.7",
-    "tensor-regression>=0.0.4",
     "simple-parsing>=0.1.5",
     "pydantic==2.7.4",
     "milatools>=0.0.18",
@@ -61,7 +60,7 @@ dev = [
     "ruff>=0.3.3",
     "pytest-benchmark>=4.0.0",
     "pytest-cov>=5.0.0",
-    "tensor-regression @ git+https://www.github.com/lebrice/tensor_regression@add-skip-creation",
+    "tensor-regression>=0.0.6",
     "pytest-testmon>=2.1.1",
     "mkdocs>=1.6.0",
     "pymdown-extensions>=10.8.1",