diff --git a/docker-compose.yml b/docker-compose.yml
index c5520fd..16e88ee 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -1,6 +1,12 @@
 name: nullinside-ui
 services:
     nullinside-ui:
+        build:
+            context: .
+            args:
+                - BUILD_ENVIRONMENT=prod
+            tags:
+                - "nullinside-ui:latest"
         logging:
             driver: loki
             options:
diff --git a/go.sh b/go.sh
index 251d9e9..3cf76b8 100644
--- a/go.sh
+++ b/go.sh
@@ -1,6 +1,6 @@
 export IP_ADDRESS=$(/sbin/ip -o -4 addr list eno2 | awk '{print $4}' | cut -d/ -f1)
 
 cp /secrets/certs/* nginx/filesystem/etc/nginx/certs
-docker build -t nullinside-ui:latest --build-arg BUILD_ENVIRONMENT=prod --build-arg IP_ADDRESS=$IP_ADDRESS .
+docker compose build --build-arg IP_ADDRESS=$IP_ADDRESS .
 docker compose down
 docker compose up -d