diff --git a/test_playlistmodel.py b/test_playlistmodel.py
index 55a8e13..76d71c7 100644
--- a/test_playlistmodel.py
+++ b/test_playlistmodel.py
@@ -202,7 +202,7 @@ def test_insert_header_row_end(monkeypatch, session):
     # Test against edit_role because display_role for headers is
     # handled differently (sets up row span)
     assert (
-        model.edit_role(model.rowCount(), playlistmodel.Col.NOTE.value, prd)
+        model.edit_role(model.rowCount() - 1, playlistmodel.Col.NOTE.value, prd)
         == note_text
     )
 
@@ -222,7 +222,7 @@ def test_insert_header_row_middle(monkeypatch, session):
     # Test against edit_role because display_role for headers is
     # handled differently (sets up row span)
     assert (
-        model.edit_role(model.rowCount(), playlistmodel.Col.NOTE.value, prd)
+        model.edit_role(model.rowCount() - 1, playlistmodel.Col.NOTE.value, prd)
         == note_text
     )