diff --git a/www/code/main.js b/www/code/main.js
index 6fffd26ec..1fb389e8a 100644
--- a/www/code/main.js
+++ b/www/code/main.js
@@ -703,9 +703,22 @@ define([
             var onAbort = config.onAbort = function (info) {
                 // inform of network disconnect
                 setEditable(false);
+                toolbar.failed();
                 Cryptpad.alert(Messages.disconnectAlert);
             };
 
+            var onConnectionChange = config.onConnectionChange = function (info) {
+                setEditable(info.state);
+                toolbar.failed();
+                if (info.state) {
+                    initializing = true;
+                    toolbar.reconnecting(info.myId);
+                    Cryptpad.findOKButton().click();
+                } else {
+                    Cryptpad.alert(Messages.disconnectAlert);
+                }
+            };
+
             var realtime = module.realtime = Realtime.start(config);
 
             editor.on('change', onLocal);