diff --git a/.bazelrc b/.bazelrc
index 5b7bc6533..f9f5cfc1b 100644
--- a/.bazelrc
+++ b/.bazelrc
@@ -1,10 +1,6 @@
 ############################################################################
 # All default build options below.
 
-# Required by OpenXLA
-# https://github.com/openxla/xla/issues/1323
-build --nocheck_visibility
-
 # Sets the default Apple platform to macOS.
 build --apple_platform_type=macos
 build --macos_minimum_os=10.14
diff --git a/jax_plugins/BUILD.bazel b/jax_plugins/BUILD.bazel
index 6e2cf6aad..08a590b09 100644
--- a/jax_plugins/BUILD.bazel
+++ b/jax_plugins/BUILD.bazel
@@ -12,6 +12,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+package(
+    default_visibility = ["//jax:internal"],
+)
+
 licenses(["notice"])
 
 load(