diff --git a/frc971/control_loops/drivetrain/BUILD b/frc971/control_loops/drivetrain/BUILD
index 16c6340..abc8aaf 100644
--- a/frc971/control_loops/drivetrain/BUILD
+++ b/frc971/control_loops/drivetrain/BUILD
@@ -11,7 +11,7 @@
     deps = [
         ":drivetrain_queue",
         "//aos/controls:replay_control_loop",
-        "//aos/linux_code:init",
+        "//aos:init",
         "//frc971/queues:gyro",
     ],
 )
diff --git a/frc971/control_loops/drivetrain/replay_drivetrain.cc b/frc971/control_loops/drivetrain/replay_drivetrain.cc
index 55e18ad..697bdec 100644
--- a/frc971/control_loops/drivetrain/replay_drivetrain.cc
+++ b/frc971/control_loops/drivetrain/replay_drivetrain.cc
@@ -1,5 +1,5 @@
 #include "aos/controls/replay_control_loop.h"
-#include "aos/linux_code/init.h"
+#include "aos/init.h"
 
 #include "frc971/control_loops/drivetrain/drivetrain.q.h"
 #include "frc971/queues/gyro.q.h"