diff --git a/flepimop/gempyor_pkg/tests/utils/test_random_distribution_sampler.py b/flepimop/gempyor_pkg/tests/utils/test_random_distribution_sampler.py
index 596d296c1..f4e26115d 100644
--- a/flepimop/gempyor_pkg/tests/utils/test_random_distribution_sampler.py
+++ b/flepimop/gempyor_pkg/tests/utils/test_random_distribution_sampler.py
@@ -14,7 +14,7 @@ class TestRandomDistributionSampler:
     def test_not_implemented_error_exception(self, distribution: str) -> None:
         with pytest.raises(
             NotImplementedError,
-            match=rf"^Unknown distribution \[received.*",
+            match=rf"^Unknown distribution \[received '{distribution}'\]",
         ):
             random_distribution_sampler(distribution)