diff --git a/projects/rocdecode/src/parser/hevc_parser.cpp b/projects/rocdecode/src/parser/hevc_parser.cpp
index 7b563bd611..e0823c6d56 100644
--- a/projects/rocdecode/src/parser/hevc_parser.cpp
+++ b/projects/rocdecode/src/parser/hevc_parser.cpp
@@ -2230,10 +2230,8 @@ int HEVCVideoParser::MarkOutputPictures() {
 
         if (!no_output_of_prior_pics_flag) {
             // Bump the remaining pictures
-            while (dpb_buffer_.num_needed_for_output) {
-                if (BumpPicFromDpb() != PARSER_OK) {
-                    return PARSER_FAIL;
-                }
+            if (FlushDpb() != PARSER_OK) {
+                return PARSER_FAIL;
             }
         }