X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=lib%2Fgraph%2Fiterator.c;fp=lib%2Fgraph%2Fiterator.c;h=a0da4e6650b9f75755b702286cd1868f56a08a3f;hb=5df26c89c52ed58a96fc1bce64d6e49d69b655ee;hp=9fc612cce5bfd1e8e1d0a48dbe7ef08b5721f3a7;hpb=8ca9c2f15b3efb3d16a44b8839b085d1f5cf903d;p=babeltrace.git

diff --git a/lib/graph/iterator.c b/lib/graph/iterator.c
index 9fc612cc..a0da4e66 100644
--- a/lib/graph/iterator.c
+++ b/lib/graph/iterator.c
@@ -48,9 +48,11 @@
 #include <babeltrace/graph/message-internal.h>
 #include <babeltrace/graph/message-event-const.h>
 #include <babeltrace/graph/message-event-internal.h>
-#include <babeltrace/graph/message-packet-const.h>
+#include <babeltrace/graph/message-packet-beginning-const.h>
+#include <babeltrace/graph/message-packet-end-const.h>
 #include <babeltrace/graph/message-packet-internal.h>
-#include <babeltrace/graph/message-stream-const.h>
+#include <babeltrace/graph/message-stream-beginning-const.h>
+#include <babeltrace/graph/message-stream-end-const.h>
 #include <babeltrace/graph/message-stream-internal.h>
 #include <babeltrace/graph/message-inactivity-internal.h>
 #include <babeltrace/graph/port-const.h>
@@ -1002,7 +1004,7 @@ int get_message_ns_from_origin(const struct bt_message *msg,
 		goto end;
 	case BT_MESSAGE_TYPE_PACKET_BEGINNING:
 	{
-		const struct bt_message_packet_beginning *pkt_msg =
+		const struct bt_message_packet *pkt_msg =
 			(const void *) msg;
 
 		clk_snapshot = pkt_msg->packet->default_beginning_cs;
@@ -1010,7 +1012,7 @@ int get_message_ns_from_origin(const struct bt_message *msg,
 	}
 	case BT_MESSAGE_TYPE_PACKET_END:
 	{
-		const struct bt_message_packet_end *pkt_msg =
+		const struct bt_message_packet *pkt_msg =
 			(const void *) msg;
 
 		clk_snapshot = pkt_msg->packet->default_end_cs;