diff --git a/app/playlists.py b/app/playlists.py
index c46526c..ed69b71 100644
--- a/app/playlists.py
+++ b/app/playlists.py
@@ -1203,14 +1203,11 @@ class PlaylistTab(QTableWidget):
             plr.track_id = track.id
             session.commit()
             # Update attributes of row
-            self.item(row, columns["userdata"].idx).setData(
-                self.ROW_TRACK_ID, track.id)
-            self.item(row, columns["start_gap"].idx).setText(
-                str(track.start_gap))
-            self.item(row, columns["title"].idx).setText(str(track.title))
-            self.item(row, columns["artist"].idx).setText(str(track.artist))
-            self.item(row, columns["duration"].idx).setText(
-                ms_to_mmss(track.duration))
+            self.item(row, USERDATA).setData(self.ROW_TRACK_ID, track.id)
+            self.item(row, START_GAP).setText(str(track.start_gap))
+            self.item(row, TITLE).setText(str(track.title))
+            self.item(row, ARTIST).setText(str(track.artist))
+            self.item(row, DURATION).setText(ms_to_mmss(track.duration))
             last_playtime = Playdates.last_played(session, track.id)
             last_played_str = get_relative_date(last_playtime)
             self.item(row, LASTPLAYED).setText(last_played_str)