diff --git a/test_playlists.py b/test_playlists.py
index 7d55b51..c6c0835 100644
--- a/test_playlists.py
+++ b/test_playlists.py
@@ -229,6 +229,8 @@ def test_get_selected_row(qtbot, monkeypatch, session):
     qtbot.mouseClick(
         playlist_tab.viewport(), Qt.LeftButton, pos=rect.center()
     )
+    row_number = playlist_tab.get_selected_row()
+    assert row_number == 0
 
 
 def test_set_next(qtbot, monkeypatch, session):
@@ -248,6 +250,9 @@ def test_set_next(qtbot, monkeypatch, session):
 
     # Add some tracks
     track1 = models.Tracks.get_by_filename(session, "isa.mp3")
+    track1_title = track1.title
+    assert track1_title
+
     playlist_tab.insert_track(session, track1)
     track2 = models.Tracks.get_by_filename(session, "mom.mp3")
     playlist_tab.insert_track(session, track2)
@@ -261,11 +266,12 @@ def test_set_next(qtbot, monkeypatch, session):
     qtbot.mouseClick(
         playlist_tab.viewport(), Qt.LeftButton, pos=rect.center()
     )
-    # qtbot.wait(10000)
+    selected_title = playlist_tab.get_selected_title()
+    assert selected_title == track1_title
+
     qtbot.keyPress(playlist_tab.viewport(), "N",
                    modifier=Qt.ControlModifier)
-    qtbot.wait(2000)
-    pass
+    qtbot.wait(1000)
 
 
 def test_kae(monkeypatch, session):
@@ -277,4 +283,3 @@ def test_kae(monkeypatch, session):
     # monkeypatch.setattr(dbconfig, "Session", session)
     # monkeypatch.setattr(models, "Session", session)
     # monkeypatch.setattr(playlists, "Session", session)
-