diff --git a/main/src/docker.rs b/main/src/docker.rs
index 5822558..fb6d1ab 100644
--- a/main/src/docker.rs
+++ b/main/src/docker.rs
@@ -64,7 +64,8 @@ fn create_image(cargo_stylus_version: &str, toolchain_version: &str) -> Result<(
     write!(
         child.stdin.as_mut().unwrap(),
         "\
-            FROM --platform=linux/amd64 offchainlabs/cargo-stylus-base:{} as base
+            ARG BUILD_PLATFORM=linux/amd64
+            FROM --platform=${{BUILD_PLATFORM}} offchainlabs/cargo-stylus-base:{} AS base
             RUN rustup toolchain install {}-x86_64-unknown-linux-gnu 
             RUN rustup default {}-x86_64-unknown-linux-gnu
             RUN rustup target add wasm32-unknown-unknown