diff --git a/src/Extension/H264.h b/src/Extension/H264.h
index 34b8dffb..9155dc01 100644
--- a/src/Extension/H264.h
+++ b/src/Extension/H264.h
@@ -68,10 +68,7 @@ public:
         _dts = dts;
         _pts = pts;
         _prefix_size = prefix_size;
-    }
-
-    CodecId getCodecId() const override{
-        return CodecH264;
+        _codec_id = CodecH264;
     }
 
     bool keyFrame() const override {
diff --git a/src/Extension/H265.h b/src/Extension/H265.h
index 94d934e8..eee14de2 100644
--- a/src/Extension/H265.h
+++ b/src/Extension/H265.h
@@ -92,10 +92,7 @@ public:
         _dts = dts;
         _pts = pts;
         _prefix_size = prefix_size;
-    }
-
-    CodecId getCodecId() const override {
-        return CodecH265;
+        _codec_id = CodecH265;
     }
 
     bool keyFrame() const override {