diff --git a/flepimop/gempyor_pkg/tests/utils/test_write_df.py b/flepimop/gempyor_pkg/tests/utils/test_write_df.py
index 8f4e18baf..e63afa2e5 100644
--- a/flepimop/gempyor_pkg/tests/utils/test_write_df.py
+++ b/flepimop/gempyor_pkg/tests/utils/test_write_df.py
@@ -30,13 +30,13 @@ def test_raises_not_implemented_error(self) -> None:
         """
         with pytest.raises(
             expected_exception=NotImplementedError,
-            match=r".*Supported extensions are `.csv` or `.parquet`.",
+            match=r"Invalid extension provided: '.txt'. Supported extensions are `.csv` or `.parquet`.",
         ) as _:
             with NamedTemporaryFile(suffix=".txt") as temp_file:
                 write_df(fname=temp_file.name, df=self.sample_df)
         with pytest.raises(
             expected_exception=NotImplementedError,
-            match=r".*Supported extensions are `.csv` or `.parquet`.",
+            match=r"Invalid extension provided: '.txt'. Supported extensions are `.csv` or `.parquet`.",
         ) as _:
             with NamedTemporaryFile(suffix=".txt") as temp_file:
                 fname = temp_file.name[:-4]