diff --git a/libavcodec/hevc_refs.c b/libavcodec/hevc_refs.c
index 24d55d15be..91347cf33f 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;