diff --git a/www/whiteboard/main.js b/www/whiteboard/main.js
index 66cc01b56..0ec3960c9 100644
--- a/www/whiteboard/main.js
+++ b/www/whiteboard/main.js
@@ -379,6 +379,17 @@ define([
         };
 
         // TODO onConnectionStateChange
+        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.common_connectionLost, undefined, true);
+            }
+        };
 
         var rt = Realtime.start(config);