diff --git a/app/stores/DocumentsStore.ts b/app/stores/DocumentsStore.ts index 8b0f0faff..9c26063fe 100644 --- a/app/stores/DocumentsStore.ts +++ b/app/stores/DocumentsStore.ts @@ -724,15 +724,17 @@ export default class DocumentsStore extends Store { ...params, ...options, }); + invariant(res?.data, "Data should be available"); const collection = this.getCollectionForDocument(res.data); await collection?.fetchDocuments({ force: true }); - const document = this.add(res.data); - this.addPolicies(res.policies); - - return document; + return runInAction("Document#update", () => { + const document = this.add(res.data); + this.addPolicies(res.policies); + return document; + }); } finally { this.isSaving = false; }