3 files changed, 9 insertions, 1 deletions
diff --git a/src/libwifi/core/misc/byteswap.h b/src/libwifi/core/misc/byteswap.h
index aa91a37..9a7d263 100644
--- a/src/libwifi/core/misc/byteswap.h
+++ b/src/libwifi/core/misc/byteswap.h

@@ -38,7 +38,7 @@
 #define be64toh(x) OSSwapBigToHostInt64(x)
 #define le64toh(x) OSSwapLittleToHostInt64(x)
 #else
-#include <byteswap.h>
+#include "byteswap.h"
 #if __BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__
 #define BYTESWAP16(x) x
diff --git a/src/libwifi/core/radiotap/platform.h b/src/libwifi/core/radiotap/platform.h
index e0ad99f..fd92481 100644
--- a/src/libwifi/core/radiotap/platform.h
+++ b/src/libwifi/core/radiotap/platform.h

@@ -35,6 +35,10 @@
 #include <sys/types.h>
 #endif
+#if defined(ESP_PLATFORM)
+#include "endian.h"
+#endif
 #ifndef le16_to_cpu
 #define le16_to_cpu le16toh
 #endif
diff --git a/src/libwifi/parse/data/eapol.c b/src/libwifi/parse/data/eapol.c
index b8e3d18..5f4a588 100644
--- a/src/libwifi/parse/data/eapol.c
+++ b/src/libwifi/parse/data/eapol.c

@@ -25,6 +25,10 @@
 #include <stdlib.h>
 #include <string.h>
+#if defined(ESP_PLATFORM)
+#include "endian.h"
+#endif
 /**
 * A libwifi_frame is deemed to be an EAPOL handshake if the following criteria is met:
 * - The frame is of type TYPE_DATA, and