diff --git a/src/libtomahawk/widgets/infowidgets/AlbumInfoWidget.cpp b/src/libtomahawk/widgets/infowidgets/AlbumInfoWidget.cpp
index 99aebceac..4b6b5a6eb 100644
--- a/src/libtomahawk/widgets/infowidgets/AlbumInfoWidget.cpp
+++ b/src/libtomahawk/widgets/infowidgets/AlbumInfoWidget.cpp
@@ -25,7 +25,7 @@
 #include "ViewManager.h"
 #include "database/Database.h"
 #include "playlist/TreeModel.h"
-#include "playlist/AlbumModel.h"
+#include "playlist/PlayableModel.h"
 #include "Source.h"
 
 #include "database/DatabaseCommand_AllTracks.h"
@@ -50,8 +50,8 @@ AlbumInfoWidget::AlbumInfoWidget( const Tomahawk::album_ptr& album, QWidget* par
     TomahawkUtils::unmarginLayout( ui->verticalLayout );
     TomahawkUtils::unmarginLayout( ui->verticalLayout_2 );
 
-    m_albumsModel = new AlbumModel( ui->albumsView );
-    ui->albumsView->setAlbumModel( m_albumsModel );
+    m_albumsModel = new PlayableModel( ui->albumsView );
+    ui->albumsView->setPlayableModel( m_albumsModel );
 
     m_tracksModel = new TreeModel( ui->tracksView );
     m_tracksModel->setMode( Mixed );
@@ -183,7 +183,7 @@ AlbumInfoWidget::gotAlbums( const QList<Tomahawk::album_ptr>& albums )
     if ( al.contains( m_album ) )
         al.removeAll( m_album );
 
-    m_albumsModel->addAlbums( al );
+    m_albumsModel->append( al );
 }
 
 
diff --git a/src/libtomahawk/widgets/infowidgets/AlbumInfoWidget.h b/src/libtomahawk/widgets/infowidgets/AlbumInfoWidget.h
index 43bcfbfa3..4234914ec 100644
--- a/src/libtomahawk/widgets/infowidgets/AlbumInfoWidget.h
+++ b/src/libtomahawk/widgets/infowidgets/AlbumInfoWidget.h
@@ -39,7 +39,7 @@
 #include "DllMacro.h"
 #include "Typedefs.h"
 
-class AlbumModel;
+class PlayableModel;
 class TreeModel;
 class OverlayButton;
 
@@ -104,7 +104,7 @@ private:
 
     Tomahawk::album_ptr m_album;
 
-    AlbumModel* m_albumsModel;
+    PlayableModel* m_albumsModel;
     TreeModel* m_tracksModel;
 
     QString m_title;