From 3db3e4f2e19fd94ef3916a03efc544d8cd54afef Mon Sep 17 00:00:00 2001
From: Rafael Moraes <50295204+glomatico@users.noreply.github.com>
Date: Thu, 26 Feb 2026 11:43:56 -0300
Subject: [PATCH] Set media_type for music videos and cleanup imports

---
 votify/interface/music_video.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/votify/interface/music_video.py b/votify/interface/music_video.py
index 5033f5f..c034331 100644
--- a/votify/interface/music_video.py
+++ b/votify/interface/music_video.py
@@ -1,11 +1,10 @@
 import asyncio
 import logging
 
-from .episode import SpotifyEpisodeInterface
-from .exceptions import VotifyDrmDisabledException
+from .enums import MediaType
+from .song import SpotifySongInterface
 from .types import MediaTags, SpotifyMedia
 from .video import SpotifyVideoInterface
-from .song import SpotifySongInterface
 
 logger = logging.getLogger(__name__)
 
@@ -47,6 +46,7 @@ class SpotifyMusicVideoInterface(SpotifyVideoInterface):
             isrc=isrc,
             label=label,
             media_id=track_data["uri"].split(":")[-1],
+            media_type=MediaType.MUSIC_VIDEO,
             producer=producer,
             rating=rating,
             url=f"https://open.spotify.com/track/{track_data['uri'].split(':')[-1]}",