diff --git a/libraries/extractor/src/main/java/androidx/media3/extractor/ts/H264Reader.java b/libraries/extractor/src/main/java/androidx/media3/extractor/ts/H264Reader.java
index e755e1a57a..952940e1b7 100644
--- a/libraries/extractor/src/main/java/androidx/media3/extractor/ts/H264Reader.java
+++ b/libraries/extractor/src/main/java/androidx/media3/extractor/ts/H264Reader.java
@@ -518,9 +518,9 @@ public final class H264Reader implements ElementaryStreamReader {
         sampleIsKeyframe = false;
         readingSample = true;
       }
+      setSampleIsKeyframe();
       // Reset NAL unit type to avoid stale state
       nalUnitType = NalUnitUtil.H264_NAL_UNIT_TYPE_UNSPECIFIED;
-      setSampleIsKeyframe();
       return sampleIsKeyframe;
     }