diff --git a/larq_compute_engine/tflite/python/BUILD b/larq_compute_engine/tflite/python/BUILD
index 3a25e789..13046dd8 100644
--- a/larq_compute_engine/tflite/python/BUILD
+++ b/larq_compute_engine/tflite/python/BUILD
@@ -1,5 +1,6 @@
 load("@org_tensorflow//tensorflow:tensorflow.bzl", "pybind_extension")
 load("@org_tensorflow//tensorflow/lite:build_def.bzl", "tflite_linkopts")
+load("@pypi//:requirements.bzl", "requirement")
 
 package(
     default_visibility = ["//visibility:public"],
@@ -35,6 +36,9 @@ py_library(
         "__init__.py",
         "interpreter_base.py",
     ],
+    deps = [
+        requirement("numpy"),
+    ],
 )
 
 py_library(
diff --git a/larq_compute_engine/tflite/tests/BUILD b/larq_compute_engine/tflite/tests/BUILD
index aea9b72c..163d402a 100644
--- a/larq_compute_engine/tflite/tests/BUILD
+++ b/larq_compute_engine/tflite/tests/BUILD
@@ -1,3 +1,5 @@
+load("@pypi//:requirements.bzl", "requirement")
+
 package(
     default_visibility = ["//visibility:public"],
     licenses = ["notice"],  # Apache 2.0
@@ -80,6 +82,7 @@ py_test(
     srcs = ["interpreter_test.py"],
     deps = [
         "//larq_compute_engine/tflite/python:interpreter",
+        requirement("numpy"),
     ],
 )