diff --git a/official/core/train_lib.py b/official/core/train_lib.py
index 3feeab973b3c5e1a8184b03e88b0aaefd1d30147..c03af32714566f6a689c706546087b94b9515aab 100644
--- a/official/core/train_lib.py
+++ b/official/core/train_lib.py
@@ -137,7 +137,7 @@ class OrbitExperimentRunner:
     return self._trainer
 
   @property
-  def checkpoint_manager(self) -> tf.train.CheckpointManager | None:
+  def checkpoint_manager(self) -> Optional[tf.train.CheckpointManager]:
     """The CheckpointManager that stores the checkpoints in a train job."""
     return self._checkpoint_manager