diff --git a/utils/notification_service.py b/utils/notification_service.py
index 43f132bf12d..b8e997ccc71 100644
--- a/utils/notification_service.py
+++ b/utils/notification_service.py
@@ -615,7 +615,10 @@ class Message:
                         pattern = r"<(https://github.com/huggingface/transformers/actions/runs/.+?/job/.+?)\|(.+?)>"
                         items = re.findall(pattern, line)
                     elif "tests/" in line:
-                        if "tests/models/" in line or "tests/quantization/" in line:
+                        # TODO: Improve the condition here.
+                        if "tests/models/" in line or (
+                            "tests/quantization/" in line and job_name == "run_quantization_torch_gpu"
+                        ):
                             model = line.split("/")[2]
                         else:
                             model = line.split("/")[1]