diff --git a/faster_whisper/transcribe.py b/faster_whisper/transcribe.py
index c1ea390b..f82e79b6 100644
--- a/faster_whisper/transcribe.py
+++ b/faster_whisper/transcribe.py
@@ -490,7 +490,7 @@ def generate_segments(
                 content_frames - seek,
                 seek_clip_end - seek,
             )
-            segment = features[:, seek : seek + segment_size]
+            segment = features[:, seek : seek + self.feature_extractor.nb_max_frames]
             segment_duration = segment_size * self.feature_extractor.time_per_frame
 
             if self.logger.isEnabledFor(logging.DEBUG):