1 files changed, 3 insertions, 3 deletions
diff --git a/src/ryuslash/org/dvdroid/DVDDataSource.java b/src/ryuslash/org/dvdroid/DVDDataSource.java
index 428ecc4..6658c8d 100644
--- a/src/ryuslash/org/dvdroid/DVDDataSource.java
+++ b/src/ryuslash/org/dvdroid/DVDDataSource.java
@@ -33,13 +33,13 @@ public class DVDDataSource
         dbHelper.close();
     }
 
-    public DVD createDVD(String title)
+    public DVD createDVD(String title, int watched, int wishlist)
     {
         ContentValues values = new ContentValues();
 
         values.put(SQLiteHelper.COLUMN_TITLE, title);
-        values.put(SQLiteHelper.COLUMN_WATCHED, 0);
-        values.put(SQLiteHelper.COLUMN_WISHLIST, 1);
+        values.put(SQLiteHelper.COLUMN_WATCHED, watched);
+        values.put(SQLiteHelper.COLUMN_WISHLIST, wishlist);
 
         long insertId = database.insert(SQLiteHelper.TABLE_DVD, null,
                                         values);