diff --git a/Dockerfile b/Dockerfile
index 9f5f54e..1e253da 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,7 +1,5 @@
-# Use the official Python 3.11 image
-# FROM --platform=arm64 python:3.11.4-bookworm
-# ARG BUILDPLATFORM
-FROM python:3.11.4-bookworm
+ARG DOCKER_PLATFORM=$TARGETPLATFORM
+FROM --platform=$DOCKER_PLATFORM python:3.11.4-bookworm
 
 ENV PYTHONBUFFERED 1