diff --git a/Dockerfile b/Dockerfile
index 389d459..71746d8 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,5 +1,4 @@
 FROM scratch
 ARG TARGETARCH
-ARG ARCH=$(if [ "$TARGETARCH" -eq "linux/arm64" ]; then echo "aarch64"; else echo "x86_64"; fi)
-COPY ./os-${ARCH}/ /
+COPY ./os-${TARGETARCH}/ /
 CMD ["/bin/sh"]
\ No newline at end of file