diff --git a/assist/Dockerfile b/assist/Dockerfile
index b316f3103..e606b008d 100644
--- a/assist/Dockerfile
+++ b/assist/Dockerfile
@@ -1,5 +1,5 @@
 #ARCH can be amd64 or arm64
-ARG ARCH=amd64
+ARG ARCH
 
 FROM --platform=linux/$ARCH node:20-alpine
 LABEL Maintainer="KRAIEM Taha Yassine<tahayk2@gmail.com>"
@@ -22,4 +22,4 @@ USER 1001
 ADD --chown=1001 https://static.openreplay.com/geoip/GeoLite2-City.mmdb  $MAXMINDDB_FILE
 
 ENTRYPOINT ["/sbin/tini", "--"]
-CMD npm start
\ No newline at end of file
+CMD npm start
diff --git a/assist/build.sh b/assist/build.sh
index 180b45898..542bd89a4 100644
--- a/assist/build.sh
+++ b/assist/build.sh
@@ -53,7 +53,7 @@ function build_api() {
     [[ $1 == "ee" ]] && {
         cp -rf ../ee/assist/* ./
     }
-    docker build -f ./Dockerfile --platform linux/${ARCH} --build-arg GIT_SHA=$git_sha -t ${DOCKER_REPO:-'local'}/assist:${image_tag} .
+    docker build -f ./Dockerfile --platform linux/${ARCH} --build-arg ARCH=$ARCH --build-arg GIT_SHA=$git_sha -t ${DOCKER_REPO:-'local'}/assist:${image_tag} .
 
     cd ../assist
     rm -rf ../${destination}