diff --git a/tests/BUILD b/tests/BUILD
index 29776452d..dd28fae50 100644
--- a/tests/BUILD
+++ b/tests/BUILD
@@ -1365,6 +1365,7 @@ exports_files(
         "python_callback_test.py",
         "shard_map_test.py",
         "transfer_guard_test.py",
+        "layout_test.py",
     ],
     visibility = jax_test_file_visibility,
 )