diff --git a/oss/build_whl.sh b/oss/build_whl.sh
index 85e1cad..d6bf018 100755
--- a/oss/build_whl.sh
+++ b/oss/build_whl.sh
@@ -91,8 +91,7 @@ function main() {
   $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
-  # Re-enable the grain import test once the new version is released.
-  # $PYTHON_BIN oss/test_import_grain.py
+  $PYTHON_BIN oss/test_import_grain.py
   $PYTHON_BIN oss/test_import_tensorflow.py
 }