diff --git a/app/models.py b/app/models.py
index 06d560b..7476bf2 100644
--- a/app/models.py
+++ b/app/models.py
@@ -486,9 +486,9 @@ class PlaylistRows(Base):
 
     @classmethod
     def get_unplayed_rows(cls, session: Session,
-                          playlist_id: int) -> List[int]:
+                          playlist_id: int) -> List["PlaylistRows]:
         """
-        For passed playlist, return a list of track rows that
+        For passed playlist, return a list of playlist rows that
         have not been played.
         """