diff --git a/main.py b/main.py
index b0f3f88..50c7202 100644
--- a/main.py
+++ b/main.py
@@ -38,6 +38,8 @@ def websocket(ws):
         if operation == "init":
             ws.send(json.dumps({"res": "ok", "perFileMax": 2147483647, "userId": 1}))
             ws.send(json.dumps({"op": "ready", "version": 11}))
+            
+            vault_id = data.get("id")
 
             hisVault = False
             for vault in database.list_vaults(data.get("token"))['vaults']:
@@ -47,7 +49,6 @@ def websocket(ws):
             if not hisVault:
                 ws.close()
 
-            vault_id = data.get("id")
             device_name = data.get("device", "unknown")
             for path, data in index["files"].items():
                 filesize = os.path.getsize(f"data/{path}")