diff --git a/assist/Dockerfile b/assist/Dockerfile
index b316f3103..102f65cb6 100644
--- a/assist/Dockerfile
+++ b/assist/Dockerfile
@@ -1,7 +1,4 @@
-#ARCH can be amd64 or arm64
-ARG ARCH=amd64
-
-FROM --platform=linux/$ARCH node:20-alpine
+FROM node:20-alpine
 LABEL Maintainer="KRAIEM Taha Yassine<tahayk2@gmail.com>"
 RUN apk add --no-cache tini
 
@@ -22,4 +19,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 f7a425285..25ccc881e 100644
--- a/assist/build.sh
+++ b/assist/build.sh
@@ -6,6 +6,7 @@
 
 # Usage: IMAGE_TAG=latest DOCKER_REPO=myDockerHubID bash build.sh <ee>
 
+ARCH=${ARCH:-amd64}
 git_sha=$(git rev-parse --short HEAD)
 image_tag=${IMAGE_TAG:-git_sha}
 check_prereq() {
@@ -51,7 +52,7 @@ function build_api() {
     [[ $1 == "ee" ]] && {
         cp -rf ../ee/assist/* ./
     }
-    docker build -f ./Dockerfile --platform linux/${ARCH:-"amd64"} --build-arg GIT_SHA=$git_sha -t ${DOCKER_REPO:-'local'}/assist:${image_tag} .
+    docker build -f ./Dockerfile --platform linux/"${ARCH:-'amd64'}" --build-arg GIT_SHA=$git_sha -t ${DOCKER_REPO:-'local'}/assist:${image_tag} .
 
     cd ../assist
     rm -rf ../${destination}