diff --git a/y2018/control_loops/python/BUILD b/y2018/control_loops/python/BUILD
index 2d2c4f1..f68939f 100644
--- a/y2018/control_loops/python/BUILD
+++ b/y2018/control_loops/python/BUILD
@@ -38,11 +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",
     ],
 )