diff --git a/main.py b/main.py
index b5b0216..b0f3f88 100644
--- a/main.py
+++ b/main.py
@@ -39,9 +39,13 @@ def websocket(ws):
             ws.send(json.dumps({"res": "ok", "perFileMax": 2147483647, "userId": 1}))
             ws.send(json.dumps({"op": "ready", "version": 11}))
 
+            hisVault = False
             for vault in database.list_vaults(data.get("token"))['vaults']:
-                if vault['id'] != vault_id:
-                    ws.close()
+                if vault['id'] == vault_id:
+                    hisVault = True
+
+            if not hisVault:
+                ws.close()
 
             vault_id = data.get("id")
             device_name = data.get("device", "unknown")