diff --git a/test/server-test.js b/test/server-test.js
index c9daf0a7..e0b2fc68 100644
--- a/test/server-test.js
+++ b/test/server-test.js
@@ -572,10 +572,13 @@ describe('Server', function() {
 
     var server = new Server(new Remote(), 'ws://localhost:5748');
 
+    server.once('disconnect', function() {
+      done();
+    });
+
     server.once('connect', function() {
       server._retryConnect = function(){
         wss.close();
-        done();
       };
       server._ws.emit('error', new Error());
     });