diff --git a/build.sh b/build.sh
index 4ab0e61..ccf6a7e 100755
--- a/build.sh
+++ b/build.sh
@@ -1,4 +1,14 @@
 #!/bin/bash
 
-version=$(<VERSION) # Read version of Dockerfile from file VERSION
-docker build "$@" -t "openworm/openworm:$version" .
+version=$(<VERSION)
+
+# Set the platform flag if we're on ARM
+arch=$(uname -m)
+if [[ "$arch" == "arm64" || "$arch" == "aarch64" ]]; then
+    platform_flag="--platform linux/amd64"
+else
+    platform_flag=""
+fi
+
+# Build the Docker image
+docker build $platform_flag "$@" -t "openworm/openworm:$version" .