diff --git a/services/screentransport/screensinkprocessor/decoder/include/image_sink_decoder.h b/services/screentransport/screensinkprocessor/decoder/include/image_sink_decoder.h
index 4e378cac863cb90f47c6e3733e4ba69638e7435f..f08b20eb4fcc3a26ddd41d117113d01b49e75abf 100644
--- a/services/screentransport/screensinkprocessor/decoder/include/image_sink_decoder.h
+++ b/services/screentransport/screensinkprocessor/decoder/include/image_sink_decoder.h
@@ -78,7 +78,7 @@ private:
     bool isDecoderReady_ = false;
     std::queue<std::shared_ptr<DataBuffer>> videoDataQueue_;
     std::queue<int32_t> bufferIndexQueue_;
-    std::shared_ptr<Media::VideoDecoder> videoDecoder_;
+    std::shared_ptr<Media::AVCodecVideoDecoder> videoDecoder_;
     std::shared_ptr<Media::AVCodecCallback> decodeVideoCallback_;
     std::weak_ptr<IImageSinkProcessorListener> imageProcessorListener_;
 };
diff --git a/services/screentransport/screensourceprocessor/encoder/include/image_source_encoder.h b/services/screentransport/screensourceprocessor/encoder/include/image_source_encoder.h
index bcf6fa55269534b6fa867a3d4b906c14671f9f0e..88c6ce5fed0d0760646781efbb778037b1e1ef6c 100644
--- a/services/screentransport/screensourceprocessor/encoder/include/image_source_encoder.h
+++ b/services/screentransport/screensourceprocessor/encoder/include/image_source_encoder.h
@@ -62,7 +62,7 @@ private:
     Media::AVCodecBufferInfo encoderBufferInfo_;
 
     sptr<Surface> videoSurface_;
-    std::shared_ptr<Media::VideoEncoder> videoEncoder_;
+    std::shared_ptr<Media::AVCodecVideoEncoder> videoEncoder_;
     std::shared_ptr<Media::AVSharedMemory> videoSharedMemory_;
     std::shared_ptr<Media::AVCodecCallback> encodeVideoCallback_;
     std::weak_ptr<IImageSourceProcessorListener> imageProcessorListener_;