X-Git-Url: http://drtracing.org/?a=blobdiff_plain;f=include%2Fbabeltrace%2Fctf-writer%2Fstream-class.h;h=f9f494366d73a6d60c64d5265cc6f52315bdc2d1;hb=c2606e2f189b382e5481ebc30f6e301ee2df61f9;hp=1cc00689a1d4fd613ba94aa4ad9b32119f66ed5f;hpb=8deee0391c182fffc2e3c41f343a008df0e42624;p=babeltrace.git

diff --git a/include/babeltrace/ctf-writer/stream-class.h b/include/babeltrace/ctf-writer/stream-class.h
index 1cc00689..f9f49436 100644
--- a/include/babeltrace/ctf-writer/stream-class.h
+++ b/include/babeltrace/ctf-writer/stream-class.h
@@ -2,8 +2,6 @@
 #define BABELTRACE_CTF_WRITER_STREAM_CLASS_H
 
 /*
- * BabelTrace - CTF Writer: Stream Class
- *
  * Copyright 2014 EfficiOS Inc.
  *
  * Author: Jérémie Galarneau <jeremie.galarneau@efficios.com>
@@ -30,8 +28,7 @@
  * http://www.efficios.com/ctf
  */
 
-#include <babeltrace/ref.h>
-#include <babeltrace/ctf-ir/stream-class.h>
+#include <babeltrace/ctf-writer/object.h>
 
 #ifdef __cplusplus
 extern "C" {
@@ -108,14 +105,14 @@ extern struct bt_ctf_clock *bt_ctf_stream_class_get_clock(
 static inline
 void bt_ctf_stream_class_get(struct bt_ctf_stream_class *stream_class)
 {
-	bt_get(stream_class);
+	bt_ctf_object_get_ref(stream_class);
 }
 
 /* Pre-2.0 CTF writer compatibility */
 static inline
 void bt_ctf_stream_class_put(struct bt_ctf_stream_class *stream_class)
 {
-	bt_put(stream_class);
+	bt_ctf_object_put_ref(stream_class);
 }
 
 #ifdef __cplusplus