diff --git a/src/__tests__/server.device.js b/src/__tests__/server.device.js
index 01fe94980..fe75c2ba3 100644
--- a/src/__tests__/server.device.js
+++ b/src/__tests__/server.device.js
@@ -55,11 +55,8 @@ test(
       // Make sure client stays connected for some time before passing test
       setTimeout(() => {
         testFinished = true;
-        if (disconnectedTooEarly) {
-          console.error('Client disconnected too early');
-        } else {
-          done();
-        }
+        expect(disconnectedTooEarly).toBe(false);
+        done();
       }, 5000);
     });
     server.addListener('removed-client', (id: string) => {