diff --git a/unit/test-gatt.c b/unit/test-gatt.c
index b60adcf..441cf34 100644
--- a/unit/test-gatt.c
+++ b/unit/test-gatt.c
#include <glib.h>
+#include "lib/bluetooth.h"
#include "lib/uuid.h"
#include "src/shared/util.h"
#include "src/shared/att.h"
diff --git a/unit/test-gattrib.c b/unit/test-gattrib.c
index 109e516..28542f3 100644
--- a/unit/test-gattrib.c
+++ b/unit/test-gattrib.c
#include <glib.h>
#include "src/shared/util.h"
+#include "lib/bluetooth.h"
#include "lib/uuid.h"
#include "attrib/att.h"
#include "attrib/gattrib.h"