diff --git a/privacy/optimizers/BUILD b/privacy/optimizers/BUILD
deleted file mode 100644
index b6ad7c3..0000000
--- a/privacy/optimizers/BUILD
+++ /dev/null
@@ -1,55 +0,0 @@
-licenses(["notice"])  # Apache 2.0
-
-py_library(
-    name = "gaussian_average_query",
-    srcs = ["gaussian_average_query.py"],
-    deps = [
-        ":private_queries",
-        "@org_tensorflow//tensorflow:tensorflow_py",
-    ],
-)
-
-py_library(
-    name = "dp_optimizers",
-    deps = [
-        ":dp_adam",
-        ":dp_gradient_descent",
-    ],
-)
-
-py_library(
-    name = "dp_adam",
-    srcs = [
-        "dp_adam.py",
-    ],
-    deps = [
-        ":gaussian_average_query",
-        "@org_tensorflow//tensorflow:tensorflow_py",
-    ],
-)
-
-py_library(
-    name = "dp_gradient_descent",
-    srcs = [
-        "dp_gradient_descent.py",
-    ],
-    deps = [
-        ":gaussian_average_query",
-        "@org_tensorflow//tensorflow:tensorflow_py",
-    ],
-)
-
-py_test(
-    name = "dp_optimizer_test",
-    srcs = ["dp_optimizer_test.py"],
-    deps = [
-        ":dp_optimizers",
-        "@absl_py//absl/testing:parameterized",
-        "@org_tensorflow//tensorflow:tensorflow_py",
-    ],
-)
-
-py_library(
-    name = "private_queries",
-    srcs = ["private_queries.py"],
-)
diff --git a/privacy/optimizers/dp_adam.py b/privacy/optimizers/dp_adam.py
index 135aa97..84aa474 100644
--- a/privacy/optimizers/dp_adam.py
+++ b/privacy/optimizers/dp_adam.py
@@ -20,7 +20,7 @@ from __future__ import print_function
 
 import tensorflow as tf
 
-import tensorflow_privacy.privacy.optimizers.gaussian_average_query as ph
+import privacy.optimizers.gaussian_average_query as ph
 
 
 class DPAdamOptimizer(tf.train.AdamOptimizer):
diff --git a/privacy/optimizers/dp_gradient_descent.py b/privacy/optimizers/dp_gradient_descent.py
index 97aa4b1..e6d5f56 100644
--- a/privacy/optimizers/dp_gradient_descent.py
+++ b/privacy/optimizers/dp_gradient_descent.py
@@ -20,7 +20,7 @@ from __future__ import print_function
 
 import tensorflow as tf
 
-import tensorflow_privacy.privacy.optimizers.gaussian_average_query as ph
+import privacy.optimizers.gaussian_average_query as ph
 
 
 class DPGradientDescentOptimizer(tf.train.GradientDescentOptimizer):
diff --git a/privacy/optimizers/dp_optimizer_test.py b/privacy/optimizers/dp_optimizer_test.py
index d351bf0..290888d 100644
--- a/privacy/optimizers/dp_optimizer_test.py
+++ b/privacy/optimizers/dp_optimizer_test.py
@@ -22,8 +22,8 @@ from absl.testing import parameterized
 import numpy as np
 import tensorflow as tf
 
-from tensorflow_privacy.privacy.optimizers import dp_adam
-from tensorflow_privacy.privacy.optimizers import dp_gradient_descent
+from privacy.optimizers import dp_adam
+from privacy.optimizers import dp_gradient_descent
 
 
 def loss(val0, val1):
diff --git a/privacy/optimizers/gaussian_average_query.py b/privacy/optimizers/gaussian_average_query.py
index 9123ffa..c6d4687 100644
--- a/privacy/optimizers/gaussian_average_query.py
+++ b/privacy/optimizers/gaussian_average_query.py
@@ -23,7 +23,7 @@ import collections
 
 import tensorflow as tf
 
-from tensorflow_privacy.privacy.optimizers import private_queries
+from privacy.optimizers import private_queries
 
 
 class GaussianAverageQuery(private_queries.PrivateAverageQuery):
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..0f57144
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1 @@
+tensorflow