X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=lib%2Fgraph%2Fcomponent-class-sink-colander.c;fp=lib%2Fgraph%2Fcomponent-class-sink-colander.c;h=e9e23d54ac065a069fa5dc917f48e12b7aa98f67;hb=07245ac23157616e3f4ff611341f18193de8e37d;hp=d69989a278fc18cbe9f39169cebceec5f0410893;hpb=26a157567b3ede2fa9e26a7f8d410aef32222be9;p=babeltrace.git

diff --git a/lib/graph/component-class-sink-colander.c b/lib/graph/component-class-sink-colander.c
index d69989a2..e9e23d54 100644
--- a/lib/graph/component-class-sink-colander.c
+++ b/lib/graph/component-class-sink-colander.c
@@ -29,6 +29,7 @@
 #include <babeltrace/graph/private-component-sink.h>
 #include <babeltrace/graph/private-port.h>
 #include <babeltrace/graph/private-connection.h>
+#include <babeltrace/graph/private-connection-notification-iterator.h>
 #include <babeltrace/graph/private-component.h>
 #include <babeltrace/graph/component-class-sink-colander-internal.h>
 #include <babeltrace/assert-internal.h>
@@ -142,8 +143,8 @@ enum bt_component_status colander_consume(
 		goto end;
 	}
 
-	notif_iter_status = bt_notification_iterator_next(
-		colander_data->notif_iter);
+	notif_iter_status = bt_private_connection_notification_iterator_next(
+		colander_data->notif_iter, &notif);
 	switch (notif_iter_status) {
 	case BT_NOTIFICATION_ITERATOR_STATUS_CANCELED:
 		status = BT_COMPONENT_STATUS_OK;
@@ -161,8 +162,6 @@ enum bt_component_status colander_consume(
 		goto end;
 	}
 
-	notif = bt_notification_iterator_get_notification(
-		colander_data->notif_iter);
 	BT_ASSERT(notif);
 
 end: