diff --git a/frc971/control_loops/python/BUILD b/frc971/control_loops/python/BUILD
index fa94d01..d172983 100644
--- a/frc971/control_loops/python/BUILD
+++ b/frc971/control_loops/python/BUILD
@@ -16,3 +16,13 @@
     '//third_party/cddlib:_cddlib.so',
   ],
 )
+
+py_test(
+  name = 'polytope_test',
+  srcs = [
+    'polytope_test.py',
+  ],
+  deps = [
+    ':controls',
+  ],
+)
diff --git a/frc971/control_loops/python/polytope_test.py b/frc971/control_loops/python/polytope_test.py
index 9a35ebe..51bf6fd 100755
--- a/frc971/control_loops/python/polytope_test.py
+++ b/frc971/control_loops/python/polytope_test.py
@@ -2,9 +2,10 @@
 
 import numpy
 from numpy.testing import *
-import polytope
 import unittest
 
+import frc971.control_loops.python.polytope as polytope
+
 __author__ = 'Austin Schuh (austin.linux@gmail.com)'
 
 def MakePoint(*args):