small fixes

1 files changed, 19 insertions, 10 deletions

diff --git a/apps/Notes/models/NoteList.js b/apps/Notes/models/NoteList.js
index 0af648a..7bad163 100644
--- a/apps/Notes/models/NoteList.js
+++ b/apps/Notes/models/NoteList.js
@@ -49,16 +49,25 @@ noteListSchema.statics.removeNote = async (_id, id) => {
 }
 
 noteListSchema.statics.updateList = async (_id, noteId, title) => {
-  const noteList = await NoteList.findOneAndUpdate(
-    { _id, 'notes.noteId': noteId },
-    {
-      $set: {
-        'notes.$.title': encrypt(title || 'No title'),
-        'notes.$.updated_at': Date.now()
-      }
-    },
-    { new: true }
-  ).lean()
+  const noteList = title
+    ? await NoteList.findOneAndUpdate(
+      { _id, 'notes.noteId': noteId },
+      {
+        $set: {
+          'notes.$.title': encrypt(title || 'No title'),
+          'notes.$.updated_at': Date.now()
+        }
+      },
+      { new: true }
+    ).lean() : await NoteList.findOneAndUpdate(
+      { _id, 'notes.noteId': noteId },
+      {
+        $set: {
+          'notes.$.updated_at': Date.now()
+        }
+      },
+      { new: true }
+    ).lean()
 
   return decryptTitles(noteList)
 }