diff --git a/libavcodec/hevc_sei.c b/libavcodec/hevc_sei.c
index 953633f4bd..631373e06f 100644
--- a/libavcodec/hevc_sei.c
+++ b/libavcodec/hevc_sei.c
@@ -550,7 +550,7 @@ static int decode_nal_sei_message(GetByteContext *gb, void *logctx, HEVCSEI *s,
 }
 
 int ff_hevc_decode_nal_sei(GetBitContext *gb, void *logctx, HEVCSEI *s,
-                           const HEVCParamSets *ps, int type)
+                           const HEVCParamSets *ps, enum HEVCNALUnitType type)
 {
     GetByteContext gbyte;
     int ret;
diff --git a/libavcodec/hevc_sei.h b/libavcodec/hevc_sei.h
index f198402333..ef987f6781 100644
--- a/libavcodec/hevc_sei.h
+++ b/libavcodec/hevc_sei.h
@@ -26,6 +26,7 @@
 #include "libavutil/buffer.h"
 
 #include "get_bits.h"
+#include "hevc.h"
 #include "sei.h"
 
 
@@ -154,7 +155,7 @@ typedef struct HEVCSEI {
 struct HEVCParamSets;
 
 int ff_hevc_decode_nal_sei(GetBitContext *gb, void *logctx, HEVCSEI *s,
-                           const struct HEVCParamSets *ps, int type);
+                           const struct HEVCParamSets *ps, enum HEVCNALUnitType type);
 
 /**
  * Reset SEI values that are stored on the Context.