diff --git a/main/uart.c b/main/uart.c
index 94a92eb..21bbaaa 100644
--- a/main/uart.c
+++ b/main/uart.c
@@ -16,7 +16,20 @@
  */
 
 #include <driver/uart.h>
-#include <driver/gpio.h>
+
+// GPIO Header
+#ifdef CONFIG_IDF_TARGET_ESP32
+#include <esp32/rom/gpio.h>
+#elif CONFIG_IDF_TARGET_ESP32C3
+#include <esp32c3/rom/gpio.h>
+#elif CONFIG_IDF_TARGET_ESP32S2
+#include <esp32s2/rom/gpio.h>
+#elif CONFIG_IDF_TARGET_ESP32S3
+#include <esp32s3/rom/gpio.h>
+#else
+#include <rom/gpio.h>
+#endif
+
 #include <esp_event.h>
 #include <esp_log.h>
 #include <string.h>