diff --git a/y2016/control_loops/python/BUILD b/y2016/control_loops/python/BUILD
index 08a34e8..49393ad 100644
--- a/y2016/control_loops/python/BUILD
+++ b/y2016/control_loops/python/BUILD
@@ -38,10 +38,14 @@
         "polydrivetrain.py",
     ],
     restricted_to = ["//tools:k8"],
+    visibility = ["//visibility:public"],
     deps = [
+        ":python_init",
         "//external:python-gflags",
         "//external:python-glog",
         "//frc971/control_loops/python:controls",
+        "//frc971/control_loops/python:drivetrain",
+        "//frc971/control_loops/python:polydrivetrain",
     ],
 )