diff --git a/test_playlistmodel.py b/test_playlistmodel.py
index 19a0839..16ca91e 100644
--- a/test_playlistmodel.py
+++ b/test_playlistmodel.py
@@ -43,8 +43,7 @@ def create_model_with_playlist_rows(
     # Create a model
     model = playlistmodel.PlaylistModel(playlist.id)
     for row in range(rows):
-        plr = model.insert_row(proposed_row_number=row, note=str(row))
-        model.playlist_rows[plr.plr_rownum] = playlistmodel.PlaylistRowData(plr)
+        model.insert_row(proposed_row_number=row, note=str(row))
 
     session.commit()
     return model