diff --git a/src/transport/af-mctp.c b/src/transport/af-mctp.c
index ae00fe4..ab2cc1c 100644
--- a/src/transport/af-mctp.c
+++ b/src/transport/af-mctp.c
@@ -3,6 +3,7 @@
 #include "container-of.h"
 #include "libpldm/pldm.h"
 #include "libpldm/transport.h"
+#include "libpldm/transport/af-mctp.h"
 #include "socket.h"
 #include "transport.h"
 
diff --git a/src/transport/mctp-demux.c b/src/transport/mctp-demux.c
index 427c52a..25b1eb3 100644
--- a/src/transport/mctp-demux.c
+++ b/src/transport/mctp-demux.c
@@ -3,6 +3,7 @@
 #include "container-of.h"
 #include "libpldm/pldm.h"
 #include "libpldm/transport.h"
+#include "libpldm/transport/mctp-demux.h"
 #include "socket.h"
 #include "transport.h"