diff --git a/src/youtube/stream_extractor.rs b/src/youtube/stream_extractor.rs
index f24670e..4840787 100644
--- a/src/youtube/stream_extractor.rs
+++ b/src/youtube/stream_extractor.rs
@@ -30,7 +30,9 @@ use crate::newpipe::NewPipe;
 use crate::stream::{
     AudioStream, DeliveryMethod, StreamInfo, StreamType, SubtitlesStream, VideoStream,
 };
-use crate::youtube::itag::{lookup as itag_lookup, ItagType, MediaFormat};
+use crate::youtube::itag::{lookup as itag_lookup, ItagType};
+#[cfg(test)]
+use crate::youtube::itag::MediaFormat;
 use crate::youtube::js::PlayerManager;
 use crate::youtube::potoken::{po_token_provider, PoTokenResult};
 use crate::youtube::stream_helper::{self, generate_content_playback_nonce};