diff --git a/grain/_src/python/dataset/transformations/BUILD b/grain/_src/python/dataset/transformations/BUILD
index aa832cc0..bba78d2a 100644
--- a/grain/_src/python/dataset/transformations/BUILD
+++ b/grain/_src/python/dataset/transformations/BUILD
@@ -392,3 +392,15 @@ py_test(
         "@pypi//numpy:pkg",
     ],
 )
+
+py_strict_test(
+    name = "autotune_sandbox_test",
+    srcs = ["autotune_sandbox_test.py"],
+    deps = [
+        ":core_transformations",
+        "//grain",
+        "//grain/_src/python:options",
+        "//grain/_src/python/experimental/autotune/python:bindings",
+        "@abseil-py//absl/testing:absltest",
+    ],
+)