diff --git a/playlist.go b/playlist.go
index 91fe216..3578b61 100644
--- a/playlist.go
+++ b/playlist.go
@@ -381,7 +381,7 @@ func (p *Playlist) Refresh() {
 	root.Walk(func(node, parent *tview.TreeNode) bool {
 
 		// to preserve previously highlighted node
-		if node.GetReference().(*AudioFile).Name == prevFileName {
+		if node.GetText() == prevFileName {
 			p.SetHighlight(node)
 			return false
 		}