X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=plugins%2Futils%2Fcounter%2Fcounter.c;h=7bf9b799dd27f9a646fa9734fefa1be1b4e54db7;hb=bb61965be5ec1bfe6977d0eca85e2e83da65af5f;hp=f9ce107fba52b935e55cf19dc9e39df87cf01657;hpb=c6578b037c207e0501396da2374a982c6c028e4f;p=babeltrace.git

diff --git a/plugins/utils/counter/counter.c b/plugins/utils/counter/counter.c
index f9ce107f..7bf9b799 100644
--- a/plugins/utils/counter/counter.c
+++ b/plugins/utils/counter/counter.c
@@ -114,7 +114,7 @@ void counter_finalize(struct bt_self_component_sink *comp)
 
 	BT_ASSERT(comp);
 	counter = bt_self_component_get_data(
-			bt_self_component_sink_borrow_self_component(comp));
+			bt_self_component_sink_as_self_component(comp));
 	BT_ASSERT(counter);
 	try_print_last(counter);
 	bt_object_put_ref(counter->notif_iter);
@@ -163,7 +163,7 @@ enum bt_self_component_status counter_init(
 	}
 
 	bt_self_component_set_data(
-		bt_self_component_sink_borrow_self_component(component),
+		bt_self_component_sink_as_self_component(component),
 		counter);
 	goto end;
 
@@ -185,7 +185,7 @@ enum bt_self_component_status counter_port_connected(
 	struct bt_self_component_port_input_notification_iterator *iterator;
 
 	counter = bt_self_component_get_data(
-		bt_self_component_sink_borrow_self_component(comp));
+		bt_self_component_sink_as_self_component(comp));
 	BT_ASSERT(counter);
 	iterator = bt_self_component_port_input_notification_iterator_create(
 		self_port);
@@ -211,7 +211,7 @@ enum bt_self_component_status counter_consume(
 	bt_notification_array notifs;
 
 	counter = bt_self_component_get_data(
-			bt_self_component_sink_borrow_self_component(comp));
+			bt_self_component_sink_as_self_component(comp));
 	BT_ASSERT(counter);
 
 	if (unlikely(!counter->notif_iter)) {