X-Git-Url: http://drtracing.org/?a=blobdiff_plain;f=lib%2Ftrace-ir%2Fevent-class.c;h=839e9a995aa94eb9963d792cd9fdb6cc7d0841ca;hb=4b70020dc7da3a97f87be1f6a0055ae2d139542f;hp=685c09c32b2153ce2d65bad7072a05e590de81f0;hpb=c2606e2f189b382e5481ebc30f6e301ee2df61f9;p=babeltrace.git

diff --git a/lib/trace-ir/event-class.c b/lib/trace-ir/event-class.c
index 685c09c3..839e9a99 100644
--- a/lib/trace-ir/event-class.c
+++ b/lib/trace-ir/event-class.c
@@ -334,7 +334,8 @@ int bt_private_event_class_set_specific_context_field_class(
 
 	bt_field_class_make_part_of_trace(field_class);
 	bt_object_put_ref(event_class->specific_context_fc);
-	event_class->specific_context_fc = bt_object_get_ref(field_class);
+	event_class->specific_context_fc = field_class;
+	bt_object_get_no_null_check(event_class->specific_context_fc);
 	bt_field_class_freeze(field_class);
 	BT_LIB_LOGV("Set event class's specific context field classe: %!+E",
 		event_class);
@@ -399,7 +400,8 @@ int bt_private_event_class_set_payload_field_class(
 
 	bt_field_class_make_part_of_trace(field_class);
 	bt_object_put_ref(event_class->payload_fc);
-	event_class->payload_fc = bt_object_get_ref(field_class);
+	event_class->payload_fc = field_class;
+	bt_object_get_no_null_check(event_class->payload_fc);
 	bt_field_class_freeze(field_class);
 	BT_LIB_LOGV("Set event class's payload field classe: %!+E", event_class);