diff --git a/tx-cucumber-tests/src/test/java/org/eclipse/tractusx/traceability/test/validator/NotificationValidator.java b/tx-cucumber-tests/src/test/java/org/eclipse/tractusx/traceability/test/validator/NotificationValidator.java
index 24d528c63e..c691e79693 100644
--- a/tx-cucumber-tests/src/test/java/org/eclipse/tractusx/traceability/test/validator/NotificationValidator.java
+++ b/tx-cucumber-tests/src/test/java/org/eclipse/tractusx/traceability/test/validator/NotificationValidator.java
@@ -65,18 +65,16 @@ public static void assertHasFields(NotificationResponse notification, Map<String
         }
 
         if (fieldsToCheck.containsKey(CLOSE_REASON.getFieldName())) {
-            assertThat(notification.getReason().close())
-                    .isEqualTo(fieldsToCheck.get(CLOSE_REASON.getFieldName()));
+            assertThat(notification.latestMessage().get().getMessage()).isEqualTo(fieldsToCheck.get(CLOSE_REASON.getFieldName()));
         }
 
         if (fieldsToCheck.containsKey(ACCEPT_REASON.getFieldName())) {
-            assertThat(notification.getReason().accept())
-                    .isEqualTo(fieldsToCheck.get(ACCEPT_REASON.getFieldName()));
+            assertThat(notification.latestMessage().get().getMessage()).isEqualTo(fieldsToCheck.get(ACCEPT_REASON.getFieldName()));
+
         }
 
         if (fieldsToCheck.containsKey(DECLINE_REASON.getFieldName())) {
-            assertThat(notification.getReason().decline())
-                    .isEqualTo(fieldsToCheck.get(DECLINE_REASON.getFieldName()));
+            assertThat(notification.latestMessage().get().getMessage()).isEqualTo(fieldsToCheck.get(DECLINE_REASON.getFieldName()));
         }
 
         if (fieldsToCheck.containsKey(SEVERITY.getFieldName())) {
diff --git a/tx-models/src/main/java/notification/response/NotificationResponse.java b/tx-models/src/main/java/notification/response/NotificationResponse.java
index 1be18a684b..b84931bb6e 100644
--- a/tx-models/src/main/java/notification/response/NotificationResponse.java
+++ b/tx-models/src/main/java/notification/response/NotificationResponse.java
@@ -100,7 +100,7 @@ public class NotificationResponse {
     private List<NotificationMessageResponse> messages;
 
 
-    public Optional<NotificationMessageResponse> latestNotification() {
+    public Optional<NotificationMessageResponse> latestMessage() {
         return messages.stream().max(Comparator.comparing(NotificationMessageResponse::getMessageDate)).stream().findFirst();
     }