diff --git a/main/tests/test_astar.cpp b/main/tests/test_astar.cpp
index 0f034c78f78..8b48f075bc6 100644
--- a/main/tests/test_astar.cpp
+++ b/main/tests/test_astar.cpp
@@ -28,12 +28,13 @@
 /* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.                */
 /*************************************************************************/
 
-#include "a_star.h"
-#include "os/os.h"
-#include <stdio.h>
-
 #include "test_astar.h"
 
+#include "core/math/a_star.h"
+#include "core/os/os.h"
+
+#include <stdio.h>
+
 namespace TestAStar {
 
 class ABCX : public AStar {
diff --git a/main/tests/test_astar.h b/main/tests/test_astar.h
index 60297420171..6458f9efb19 100644
--- a/main/tests/test_astar.h
+++ b/main/tests/test_astar.h
@@ -31,7 +31,7 @@
 #ifndef TEST_ASTAR_H
 #define TEST_ASTAR_H
 
-#include "os/main_loop.h"
+#include "core/os/main_loop.h"
 
 namespace TestAStar {