diff --git a/server/routes/api/documents.ts b/server/routes/api/documents.ts
index ab1e389da..f83de06fa 100644
--- a/server/routes/api/documents.ts
+++ b/server/routes/api/documents.ts
@@ -863,6 +863,7 @@ router.post("documents.update", auth(), async (ctx) => {
     userId: user.id,
     includeState: true,
   });
+  collection = document?.collection;
   authorize(user, "update", document);
 
   if (publish) {