diff --git a/server/api/documents.js b/server/api/documents.js
index cf525861..e482c79d 100644
--- a/server/api/documents.js
+++ b/server/api/documents.js
@@ -221,6 +221,7 @@ router.post('documents.update', auth(), async ctx => {
   const collection = await Collection.findById(document.atlasId);
   if (collection.type === 'atlas') {
     await collection.updateDocument(document);
+    document.collection = collection;
   }
 
   ctx.body = {