diff --git a/app/src/main/java/com/aritra/notify/ui/screens/notes/addEditScreen/AddEditViewModel.kt b/app/src/main/java/com/aritra/notify/ui/screens/notes/addEditScreen/AddEditViewModel.kt
index f59d321..884568c 100644
--- a/app/src/main/java/com/aritra/notify/ui/screens/notes/addEditScreen/AddEditViewModel.kt
+++ b/app/src/main/java/com/aritra/notify/ui/screens/notes/addEditScreen/AddEditViewModel.kt
@@ -104,14 +104,7 @@ class AddEditViewModel @Inject constructor(
             }
         }
     }
-    fun updateReminderDateTime(dateTime: LocalDateTime?) {
-        _note.update {
-            it.copy(
-                reminderDateTime = dateTime,
-                isReminded = false
-            )
-        }
-    }
+
     fun updateNote(
         title: String,
         description: String,
@@ -142,7 +135,8 @@ class AddEditViewModel @Inject constructor(
             newNote.copy(
                 title = title,
                 note = description,
-                dateTime = Date()
+                dateTime = Date(),
+                checklist = checklist
                 // if the image has not been modified, use the old image uri
 //                image = if (oldNote.image == newNote.image) {
 //                    oldNote.image
@@ -163,4 +157,13 @@ class AddEditViewModel @Inject constructor(
             onSuccess(true)
         }
     }
+
+    fun updateReminderDateTime(dateTime: LocalDateTime?) {
+        _note.update {
+            it.copy(
+                reminderDateTime = dateTime,
+                isReminded = false
+            )
+        }
+    }
 }