diff --git a/libraries/transformer/src/androidTest/java/androidx/media3/transformer/mh/VideoTimestampConsistencyTest.java b/libraries/transformer/src/androidTest/java/androidx/media3/transformer/mh/VideoTimestampConsistencyTest.java
index 66fd3a111e..348620d8dc 100644
--- a/libraries/transformer/src/androidTest/java/androidx/media3/transformer/mh/VideoTimestampConsistencyTest.java
+++ b/libraries/transformer/src/androidTest/java/androidx/media3/transformer/mh/VideoTimestampConsistencyTest.java
@@ -367,20 +367,20 @@ public class VideoTimestampConsistencyTest {
   private void compareTimestamps(
       List<EditedMediaItem> mediaItems, List<Long> expectedTimestamps, boolean containsImage)
       throws Exception {
+    ImmutableList<Long> timestampsFromTransformer = getTimestampsFromTransformer(mediaItems);
+    assertThat(timestampsFromTransformer).isEqualTo(expectedTimestamps);
+
     ImmutableList<Long> timestampsFromCompositionPlayer =
         getTimestampsFromCompositionPlayer(mediaItems);
-    ImmutableList<Long> timestampsFromTransformer = getTimestampsFromTransformer(mediaItems);
-
-    assertThat(timestampsFromCompositionPlayer).isEqualTo(timestampsFromTransformer);
+    assertThat(timestampsFromCompositionPlayer).isEqualTo(expectedTimestamps);
 
     if (!containsImage) {
       // ExoPlayer doesn't support image playback with effects.
       ImmutableList<Long> timestampsFromExoPlayer =
           getTimestampsFromExoPlayer(
               Lists.transform(mediaItems, editedMediaItem -> editedMediaItem.mediaItem));
-      assertThat(timestampsFromCompositionPlayer).isEqualTo(timestampsFromExoPlayer);
+      assertThat(timestampsFromExoPlayer).isEqualTo(expectedTimestamps);
     }
-    assertThat(timestampsFromTransformer).isEqualTo(expectedTimestamps);
   }
 
   private ImmutableList<Long> getTimestampsFromTransformer(List<EditedMediaItem> editedMediaItems)