diff --git a/esp32/test/gait_performance.cpp b/esp32/test/gait_performance.cpp
index 465ebd9..22cbb08 100644
--- a/esp32/test/gait_performance.cpp
+++ b/esp32/test/gait_performance.cpp
@@ -1,6 +1,7 @@
 #include <Arduino.h>
 #include <unity.h>
-#include "Gait/GaitState.h"
+#include "gait/state.h"
+#include "gait/bezier_state.h"
 
 void test_gaitPlanner_calculateStep_time() {
     BezierState gaitPlanner;