diff --git a/.gitea/workflows/test.yml b/.gitea/workflows/test.yml
index 3c5a824..d3df873 100644
--- a/.gitea/workflows/test.yml
+++ b/.gitea/workflows/test.yml
@@ -49,6 +49,7 @@ jobs:
         run: |
           echo "API_URL=http://api:3000" >> .env
           echo "CDN_URL=http://cdn:4000" >> .env
+          echo "WS_URL=ws://api:3000" >> .env
 
       - name: Run Vitest
         run: |
diff --git a/src/core/environment.ts b/src/core/environment.ts
index 8d15141..862e412 100644
--- a/src/core/environment.ts
+++ b/src/core/environment.ts
@@ -1,6 +1,7 @@
 export interface SDKConfig {
     apiUrl: string;
     cdnUrl: string;
+    wsUrl: string;
 }
 
 const isNode =
@@ -22,6 +23,7 @@ const getEnv = (key: string): string | undefined => {
 const DefaultEnvironment: SDKConfig = {
     apiUrl: getEnv('API_URL') ?? "https://api.chatenium.hu",
     cdnUrl: getEnv('CDN_URL') ?? "https://cdn.chatenium.hu",
+    cdnUrl: getEnv('WS_URL') ?? "wss://api.chatenium.hu",
 };
 
 let currentConfig: SDKConfig = { ...DefaultEnvironment };
diff --git a/src/core/webSocketHandler.ts b/src/core/webSocketHandler.ts
index 30616cf..16168d2 100644
--- a/src/core/webSocketHandler.ts
+++ b/src/core/webSocketHandler.ts
@@ -40,7 +40,7 @@ export class WebSocketHandler {
             const resp = await client.post<WSMakeTokenResp>("v2/ws/makeToken", <WSMakeTokenReq>{
                 userid: userid,
             });
-            this.connection = new WebSocket(`${environment.get().apiUrl}/v2/ws?userid=${userid}&access_token=${resp.data.token}`)
+            this.connection = new WebSocket(`${environment.get().wsUrl}/v2/ws?userid=${userid}&access_token=${resp.data.token}`)
             console.log("Connected to websocket successfully")
             this.startListening()
             return