diff --git a/python/ray/tests/BUILD b/python/ray/tests/BUILD
index 2d9c569aa..c5c82c3d2 100644
--- a/python/ray/tests/BUILD
+++ b/python/ray/tests/BUILD
@@ -47,6 +47,7 @@ py_test_module_list(
     "test_stress.py",
     "test_stress_sharded.py",
     "test_unreconstructable_errors.py",
+    "test_tensorflow.py",
   ],
   size = "medium",
   extra_srcs = SRCS,
@@ -78,7 +79,6 @@ py_test_module_list(
     "test_ray_init.py",
     "test_serialization.py",
     "test_tempfile.py",
-    "test_tensorflow.py",
     "test_webui.py",
   ],
   size = "small",