diff --git a/app.dockerfile b/app.dockerfile
index 488e64b..5adf61e 100644
--- a/app.dockerfile
+++ b/app.dockerfile
@@ -1,9 +1,7 @@
 FROM node:20.18.0-alpine
 
-ARG NEXT_PUBLIC_WS_URL=ws://127.0.0.1:3001
-ARG NEXT_PUBLIC_API_URL=http://127.0.0.1:3001/api
-ENV NEXT_PUBLIC_WS_URL=${NEXT_PUBLIC_WS_URL}
-ENV NEXT_PUBLIC_API_URL=${NEXT_PUBLIC_API_URL}
+ENV NEXT_PUBLIC_WS_URL=ws://127.0.0.1:3001
+ENV NEXT_PUBLIC_API_URL=http://127.0.0.1:3001/api
 
 WORKDIR /home/perplexica
 
diff --git a/docker-compose.yaml b/docker-compose.yaml
index a0e1d73..e3b903d 100644
--- a/docker-compose.yaml
+++ b/docker-compose.yaml
@@ -34,10 +34,10 @@ services:
     build:
       context: .
       dockerfile: app.dockerfile
-      args:
-        - NEXT_PUBLIC_API_URL=http://127.0.0.1:3001/api
-        - NEXT_PUBLIC_WS_URL=ws://127.0.0.1:3001
     image: itzcrazykns1337/perplexica-frontend:main
+    environment:
+      - NEXT_PUBLIC_API_URL=http://127.0.0.1:3001/api
+      - NEXT_PUBLIC_WS_URL=ws://127.0.0.1:3001
     depends_on:
       - perplexica-backend
     ports: