diff --git a/src/middleware/Albums.js b/src/middleware/Albums.js
index 3ca4558..a502bd3 100644
--- a/src/middleware/Albums.js
+++ b/src/middleware/Albums.js
@@ -191,10 +191,11 @@ class Albums extends Pages {
     async patchOne() {
         const { itemId: _id } = this.req.params;
         const { _id: User } = this.req.user;
-        const album = await AlbumsModel.findOne({
+        const query = {
             _id,
             User,
-        });
+        };
+        const album = await AlbumsModel.findOne(query);
 
         if (!album) {
             throw new ErrorEvent(
@@ -206,9 +207,7 @@ class Albums extends Pages {
 
         const values = await getAlbumDetails(album.discogsId);
 
-        await album.updateOne(values);
-
-        return album;
+        return AlbumsModel.findOneAndUpdate(query, values, { new: true });
     }
 
     /**