1 files changed, 7 insertions, 2 deletions
diff --git a/src/ryuslash/org/dvdroid/DVDroidActivity.java b/src/ryuslash/org/dvdroid/DVDroidActivity.java
index e510a7b..533bc62 100644
--- a/src/ryuslash/org/dvdroid/DVDroidActivity.java
+++ b/src/ryuslash/org/dvdroid/DVDroidActivity.java
@@ -18,6 +18,11 @@ public class DVDroidActivity extends ListActivity
 {
     private DVDDataSource datasource;
 
+    private List<DVD> getData()
+    {
+        return datasource.getCollection();
+    }
+
     @Override
     protected void onActivityResult(int requestCode,
                                     int resultCode,
@@ -27,7 +32,7 @@ public class DVDroidActivity extends ListActivity
         datasource.open();
 
         ArrayAdapter<DVD> adapter = (ArrayAdapter<DVD>)getListAdapter();
-        List<DVD> dvds = datasource.getWishlist();
+        List<DVD> dvds = getData();
 
         adapter.clear();
         adapter.addAll(dvds);
@@ -44,7 +49,7 @@ public class DVDroidActivity extends ListActivity
         datasource = new DVDDataSource(this);
         datasource.open();
 
-        List<DVD> values = datasource.getWishlist();
+        List<DVD> values = getData();
 
         ArrayAdapter<DVD> adapter =
             new ArrayAdapter<DVD>(this,