X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=lib%2Fgraph%2Fcomponent-class.c;fp=lib%2Fgraph%2Fcomponent-class.c;h=5a306fb80019ad152f1ed97c83cd7819c1af5ff7;hb=65300d60e4b4f167e5fc8f584677757ce09a3844;hp=7ce6667fc6c03957edd65fc994e13fa249410b65;hpb=07208d85d16704b304b371897bb7acc995fcc481;p=babeltrace.git

diff --git a/lib/graph/component-class.c b/lib/graph/component-class.c
index 7ce6667f..5a306fb8 100644
--- a/lib/graph/component-class.c
+++ b/lib/graph/component-class.c
@@ -31,7 +31,7 @@
 
 #include <babeltrace/compiler-internal.h>
 #include <babeltrace/graph/component-class-internal.h>
-#include <babeltrace/ref.h>
+#include <babeltrace/object.h>
 #include <babeltrace/types.h>
 #include <babeltrace/assert-internal.h>
 #include <glib.h>
@@ -114,7 +114,7 @@ int bt_component_class_init(struct bt_component_class *class,
 	goto end;
 
 error:
-	BT_PUT(class);
+	BT_OBJECT_PUT_REF_AND_RESET(class);
 	ret = -1;
 
 end: