diff --git a/y2014/control_loops/drivetrain/BUILD b/y2014/control_loops/drivetrain/BUILD
index 1c70601..14003b1 100644
--- a/y2014/control_loops/drivetrain/BUILD
+++ b/y2014/control_loops/drivetrain/BUILD
@@ -19,6 +19,9 @@
   srcs = [
     'drivetrain.q',
   ],
+  deps = [
+    '//aos/common/controls:control_loop_queues',
+  ],
 )
 
 cc_library(
@@ -27,6 +30,10 @@
     'polydrivetrain_dog_motor_plant.cc',
     'drivetrain_dog_motor_plant.cc',
   ],
+  hdrs = [
+    'polydrivetrain_dog_motor_plant.h',
+    'drivetrain_dog_motor_plant.h',
+  ],
   deps = [
     '//frc971/control_loops:state_feedback_loop',
   ],
@@ -38,6 +45,10 @@
     'drivetrain.cc',
     'polydrivetrain_cim_plant.cc',
   ],
+  hdrs = [
+    'drivetrain.h',
+    'polydrivetrain_cim_plant.h',
+  ],
   deps = [
     ':drivetrain_queue',
     '//aos/common/controls:control_loop',
@@ -52,7 +63,7 @@
   ],
 )
 
-cc_binary(
+cc_test(
   name = 'drivetrain_lib_test',
   srcs = [
     'drivetrain_lib_test.cc',