X-Git-Url: http://git.efficios.com/?p=babeltrace.git;a=blobdiff_plain;f=include%2Fbabeltrace%2Fgraph%2Fnotification-stream-internal.h;h=5c0f72260c3c50e1ce2d8bfd07c2ec38ad8e3eab;hp=df1b3dcb3a565d95fd516450b27a9c0bad7d8f11;hb=f6ccaed94e575af57fe6bf38154771bee4871a2a;hpb=464ebc311d460b29f681703aea0aa00eef9e6475

diff --git a/include/babeltrace/graph/notification-stream-internal.h b/include/babeltrace/graph/notification-stream-internal.h
index df1b3dcb..5c0f7226 100644
--- a/include/babeltrace/graph/notification-stream-internal.h
+++ b/include/babeltrace/graph/notification-stream-internal.h
@@ -30,6 +30,7 @@
 #include <babeltrace/compiler-internal.h>
 #include <babeltrace/ctf-ir/packet.h>
 #include <babeltrace/graph/notification-internal.h>
+#include <babeltrace/assert-internal.h>
 
 struct bt_notification_stream_begin {
 	struct bt_notification parent;
@@ -49,7 +50,7 @@ struct bt_stream *bt_notification_stream_begin_borrow_stream(
 		container_of(notif,
 			struct bt_notification_stream_begin, parent);
 
-	assert(notif_stream_begin);
+	BT_ASSERT(notif_stream_begin);
 	return notif_stream_begin->stream;
 }
 
@@ -61,7 +62,7 @@ struct bt_stream *bt_notification_stream_end_borrow_stream(
 		container_of(notif,
 			struct bt_notification_stream_end, parent);
 
-	assert(notif_stream_end);
+	BT_ASSERT(notif_stream_end);
 	return notif_stream_end->stream;
 }