diff --git a/wax-prosemirror-services/src/YjsService/YjsService.js b/wax-prosemirror-services/src/YjsService/YjsService.js
index 66e0d383ebc69573701d7b4fa808471d1cd4db05..d203214bd99d872f0ab2ae497da616b63a9b1c13 100644
--- a/wax-prosemirror-services/src/YjsService/YjsService.js
+++ b/wax-prosemirror-services/src/YjsService/YjsService.js
@@ -23,8 +23,8 @@ class YjsService extends Service {
     if (!configProvider || !configYdoc) {
       ydoc = new Y.Doc();
       provider = new WebsocketProvider(connectionUrl, docIdentifier, ydoc);
-      this.app.context.setOption({ currentYdoc: ydoc });
     }
+    this.app.context.setOption({ currentYdoc: ydoc });
 
     provider.on('sync', args => {
       console.log({ sync: args });