diff --git a/MODULE.bazel b/MODULE.bazel
index b3d252d..51ee0a6 100644
--- a/MODULE.bazel
+++ b/MODULE.bazel
@@ -21,15 +21,15 @@ module(
     repo_name = "com_google_array_record",
 )
 
-bazel_dep(name = "rules_proto", version = "7.0.2")
-bazel_dep(name = "rules_python", version = "0.37.0")
-bazel_dep(name = "platforms", version = "0.0.10")
-bazel_dep(name = "protobuf", version = "28.3")
-bazel_dep(name = "googletest", version = "1.15.2")
-bazel_dep(name = "abseil-cpp", version = "20240722.0")
+bazel_dep(name = "rules_proto", version = "7.1.0")
+bazel_dep(name = "rules_python", version = "1.4.1")
+bazel_dep(name = "platforms", version = "1.0.0")
+bazel_dep(name = "protobuf", version = "31.1")
+bazel_dep(name = "googletest", version = "1.17.0")
+bazel_dep(name = "abseil-cpp", version = "20250814.1")
 bazel_dep(name = "abseil-py", version = "2.1.0")
 bazel_dep(name = "eigen", version = "3.4.0.bcr.3")
-bazel_dep(name = "riegeli", version = "0.0.0-20241218-3385e3c")
+bazel_dep(name = "riegeli", version = "0.0.0-20250717-5b2e77e")
 bazel_dep(name = "pybind11_bazel", version = "2.12.0")
 
 SUPPORTED_PYTHON_VERSIONS = [
diff --git a/oss/build_whl.sh b/oss/build_whl.sh
index d6bf018..4020099 100755
--- a/oss/build_whl.sh
+++ b/oss/build_whl.sh
@@ -90,7 +90,7 @@ function main() {
   $PYTHON_BIN -m pip install ${OUTPUT_DIR}/all_dist/array_record*.whl
   $PYTHON_BIN -c 'import array_record'
   $PYTHON_BIN -c 'from array_record.python import array_record_data_source'
-  $PYTHON_BIN -m pip install jax tensorflow>=2.20.0 grain
+  $PYTHON_BIN -m pip install jax tensorflow>=2.20.0 grain h5py
   $PYTHON_BIN oss/test_import_grain.py
   $PYTHON_BIN oss/test_import_tensorflow.py
 }