diff --git a/lib/client/edit.js b/lib/client/edit.js
index c65562ce..430a1159 100644
--- a/lib/client/edit.js
+++ b/lib/client/edit.js
@@ -47,7 +47,7 @@ var CloudCmd, Util, DOM, CloudFunc, io, ace, DiffProto, diff_match_patch, Zip, M
                 load,
                 Edit.show.bind(null, callback),
                 function() {
-                    initSocket();
+                    DOM.loadSocket(initSocket);
                 }
             ]);
         }