diff --git a/traffic_lights/lib/lib.cpp b/traffic_lights/src/lib.cpp
similarity index 88%
rename from traffic_lights/lib/lib.cpp
rename to traffic_lights/src/lib.cpp
index a9476d6..dc8d2b8 100644
--- a/traffic_lights/lib/lib.cpp
+++ b/traffic_lights/src/lib.cpp
@@ -2,7 +2,8 @@
 // Created by Gabriel Augendre on 27/04/2021.
 //
 
-#include "lib.h"
+#include <lib.h>
+#include <Arduino.h>
 
 void waitForButton() {
     while (digitalRead(BUTTON) == HIGH) {}
diff --git a/traffic_lights/src/main.cpp b/traffic_lights/src/main.cpp
index 1c1a056..31045a6 100644
--- a/traffic_lights/src/main.cpp
+++ b/traffic_lights/src/main.cpp
@@ -1,5 +1,5 @@
 #include <Arduino.h>
-#include "lib.h"
+#include <lib.h>
 
 void setup() {
     pinMode(GREEN, OUTPUT);