diff --git a/src/save.py b/src/save.py
index cc7cf44..df63483 100644
--- a/src/save.py
+++ b/src/save.py
@@ -3,11 +3,11 @@ from pathlib import Path
 
 from src.config import Download
 from src.metadata import SongMetadata
-from src.models import PlaylistMeta
+from src.models import PlaylistInfo
 from src.utils import ttml_convent_to_lrc, get_song_name_and_dir_path, get_suffix
 
 
-def save(song: bytes, codec: str, metadata: SongMetadata, config: Download, playlist: PlaylistMeta = None):
+def save(song: bytes, codec: str, metadata: SongMetadata, config: Download, playlist: PlaylistInfo = None):
     song_name, dir_path = get_song_name_and_dir_path(codec, config, metadata, playlist)
     if not dir_path.exists() or not dir_path.is_dir():
         os.makedirs(dir_path.absolute())