diff --git a/src/main.rs b/src/main.rs
index 501a043..b5e2379 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,6 +1,3 @@
-use std::time::Duration;
-
-use aspotify::Client;
 use teloxide::prelude::*;
 
 use spotify::SpotifyKind::Track;
diff --git a/src/spotify/mod.rs b/src/spotify/mod.rs
index e56f36b..cb54cb6 100644
--- a/src/spotify/mod.rs
+++ b/src/spotify/mod.rs
@@ -39,17 +39,17 @@ pub fn get_spotify_client() -> Box<Client> {
     spotify_client
 }
 
-struct TrackInfo {
-    name: String,
-    artists: Vec<String>,
-    duration: Duration,
+pub struct TrackInfo {
+    pub(crate) name: String,
+    pub(crate) artists: Vec<String>,
+    pub(crate) duration: Duration,
 }
 
-struct AlbumInfo {
+pub struct AlbumInfo {
     pub(crate) name: String,
     pub(crate) artists: Vec<String>,
     pub(crate) genres: Vec<String>,
-    songs: Vec<TrackInfo>,
+    pub(crate) songs: Vec<TrackInfo>,
 }
 
 pub async fn get_spotify_track(spotify: Box<Client>, id: &String) -> Option<TrackInfo> {