X-Git-Url: http://git.efficios.com/?p=babeltrace.git;a=blobdiff_plain;f=lib%2Fgraph%2Fnotification%2Fdiscarded-elements.c;h=69b54035987df9384a5df865fa9f4e28ec35802f;hp=ee4e8a5928b14d6279213960e216b2357732f4a5;hb=126215b484c9df27d192690e0dba2fb9f630e10a;hpb=0684124bf979ac663c63ed68f9969cf5eacf2cd6

diff --git a/lib/graph/notification/discarded-elements.c b/lib/graph/notification/discarded-elements.c
index ee4e8a59..69b54035 100644
--- a/lib/graph/notification/discarded-elements.c
+++ b/lib/graph/notification/discarded-elements.c
@@ -115,8 +115,7 @@ bt_notification_discarded_elements_get_begin_clock_value(
 		goto end;
 	}
 
-	if (bt_notification_get_type(notification) !=
-			BT_NOTIFICATION_TYPE_DISCARDED_EVENTS) {
+	if (bt_notification_get_type(notification) != type) {
 		BT_LOGW("Invalid parameter: notification has not the expected type: "
 			"addr%p, expected-type=%s, notif-type=%s",
 			notification, bt_notification_type_string(type),
@@ -147,8 +146,7 @@ bt_notification_discarded_elements_get_end_clock_value(
 		goto end;
 	}
 
-	if (bt_notification_get_type(notification) !=
-			BT_NOTIFICATION_TYPE_DISCARDED_EVENTS) {
+	if (bt_notification_get_type(notification) != type) {
 		BT_LOGW("Invalid parameter: notification has not the expected type: "
 			"addr%p, expected-type=%s, notif-type=%s",
 			notification, bt_notification_type_string(type),
@@ -178,8 +176,7 @@ int64_t bt_notification_discarded_elements_get_count(
 		goto end;
 	}
 
-	if (bt_notification_get_type(notification) !=
-			BT_NOTIFICATION_TYPE_DISCARDED_EVENTS) {
+	if (bt_notification_get_type(notification) != type) {
 		BT_LOGW("Invalid parameter: notification has not the expected type: "
 			"addr%p, expected-type=%s, notif-type=%s",
 			notification, bt_notification_type_string(type),
@@ -209,8 +206,7 @@ struct bt_ctf_stream *bt_notification_discarded_elements_get_stream(
 		goto end;
 	}
 
-	if (bt_notification_get_type(notification) !=
-			BT_NOTIFICATION_TYPE_DISCARDED_EVENTS) {
+	if (bt_notification_get_type(notification) != type) {
 		BT_LOGW("Invalid parameter: notification has not the expected type: "
 			"addr%p, expected-type=%s, notif-type=%s",
 			notification, bt_notification_type_string(type),