diff --git a/libavcodec/hevc_refs.c b/libavcodec/hevc_refs.c
index 24d55d15bec14a2bda0c3f784fc993a93715165b..91347cf33fe1db9eea4f557e7f3296e9c2dd8a0e 100644
--- a/libavcodec/hevc_refs.c
+++ b/libavcodec/hevc_refs.c
@@ -207,6 +207,9 @@ int ff_hevc_output_frame(HEVCContext *s, AVFrame *out, int flush)
                           (frame->window.top_offset   >> vshift) * dst->linesize[i];
                 dst->data[i] += off;
             }
+
+            out->key_frame = out->pict_type == AV_PICTURE_TYPE_I;
+
             av_log(s->avctx, AV_LOG_DEBUG,
                    "Output frame with POC %d.\n", frame->poc);
             return 1;