X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=plugins%2Fsource.c;h=2dba33bd432a6bf958ffbf619d3d1af4c11f2e9a;hb=b8a068016f0953440df334b147db75e861e7f2ac;hp=38fc3110801662974223dba48a4609e18e621490;hpb=dafd80ce71a4f617deaac7084edb0c03450b3259;p=babeltrace.git

diff --git a/plugins/source.c b/plugins/source.c
index 38fc3110..2dba33bd 100644
--- a/plugins/source.c
+++ b/plugins/source.c
@@ -26,6 +26,7 @@
  * SOFTWARE.
  */
 
+#include <babeltrace/ref.h>
 #include <babeltrace/compiler.h>
 #include <babeltrace/plugin/source-internal.h>
 #include <babeltrace/plugin/component-internal.h>
@@ -102,6 +103,6 @@ struct bt_notification_iterator *bt_plugin_source_create_iterator(
 end:
 	return iterator;
 error:
-	bt_notification_iterator_put(iterator);
-	return NULL;
+	BT_PUT(iterator);
+	return iterator;
 }