diff --git a/export/orbax/export/bundle_test.py b/export/orbax/export/bundle_test.py
index ec64388b2..96af6bfff 100644
--- a/export/orbax/export/bundle_test.py
+++ b/export/orbax/export/bundle_test.py
@@ -28,6 +28,7 @@
 
 """Test suite for model bundle execution."""
 
+from collections.abc import Sequence
 import os
 
 from absl.testing import absltest