diff --git a/Inc/DS_Button.h b/Inc/DS_Button.h
index f59e61a..afed754 100644
--- a/Inc/DS_Button.h
+++ b/Inc/DS_Button.h
@@ -2,7 +2,15 @@
 #define DS_BUTTON_H
 
 #include <stdbool.h>
-#include "stm32f0xx_hal.h"
+
+#ifdef STM32G030xx 
+    #include "stm32g0xx_hal.h"
+#endif
+
+#ifdef STM32F030xx 
+    #include "stm32f0xx_hal.h"
+#endif
+
 
 typedef struct DS_Button
 {