diff --git a/tests/trainer/test_checkpoint.py b/tests/trainer/test_checkpoint.py
index a24f5d3c91..7bf8dc59fe 100644
--- a/tests/trainer/test_checkpoint.py
+++ b/tests/trainer/test_checkpoint.py
@@ -864,6 +864,9 @@ def test_load_remote_checkpoint(self, device, tmp_path: pathlib.Path, load_weigh
             device=device,
         )
 
+        # TODO(GRT-2735): Update remote checkpoint with iteration.
+        trainer_2.state.timestamp._epoch_in_iteration = Time.from_input(2, TimeUnit.EPOCH)
+
         # check weights loaded properly
         self._assert_weights_equivalent(
             trainer_1.state.model,