diff --git a/libsaria/sources/__init__.py b/libsaria/sources/__init__.py
index 2194c647..232335b3 100644
--- a/libsaria/sources/__init__.py
+++ b/libsaria/sources/__init__.py
@@ -13,8 +13,6 @@ controls = libsaria.controls
 cur_source = None
 cur_id = None
 
-LIBRARY, PLAYLIST, QUEUE = range(3)
-
 class Source:
 	def __init__(self):
 		self.get_attrs  = None
@@ -70,22 +68,16 @@ def next():
 		cur_id = id
 		return library.get_attrs(cur_id, "filepath")
 
-def list_source(source, *attrs):
+def list_source(list_func, *attrs):
 	ret = []
-	list_func = None
 	get_attrs = library.get_attrs
 	append    = ret.append
-	if source == PLAYLIST:
-		list_func = playlist.list_ids
-	elif source == QUEUE:
-		list_func = queue.list_ids
-	if list_func != None:
-		for id in list_func():
-			append(get_attrs(id, *attrs))
+	for id in list_func():
+		append(get_attrs(id, *attrs))
 	return ret
 
-def list_queue(*attrs):
-	return list_source(QUEUE, *attrs)
-
 def list_playlist(*attrs):
-	return list_source(PLAYLIST, *attrs)
+	return list_source(playlist.list_ids, *attrs)
+
+def list_queue(*attrs):
+	return list_source(queue.list_ids, *attrs)