diff --git a/.gitea/workflows/build-jupyter-python.yaml b/.gitea/workflows/build-jupyter-python.yaml
index cb47a62..0682af6 100644
--- a/.gitea/workflows/build-jupyter-python.yaml
+++ b/.gitea/workflows/build-jupyter-python.yaml
@@ -29,7 +29,7 @@ jobs:
         uses: docker/build-push-action@v5.3.0
         with:
           context: "{{defaultContext}}:jupyter-python"
-          platforms: linux/amd64
+          platforms: linux/amd64,linux/arm64
           push: true
           pull: true
           tags: gitea.woggioni.net/woggioni/jupyter-python:latest
diff --git a/jupyter-python/Dockerfile b/jupyter-python/Dockerfile
index 0325470..e0061a1 100644
--- a/jupyter-python/Dockerfile
+++ b/jupyter-python/Dockerfile
@@ -2,7 +2,7 @@ FROM debian:stable AS build
 RUN --mount=type=cache,target=/var/cache/apt,sharing=locked \
     --mount=type=cache,target=/var/lib/apt,sharing=locked \
     apt update &&\
-    apt install -y python3 python3-dev pipx gcc g++ linux-headers-amd64 libffi-dev cargo cmake autoconf swig ninja-build patchelf meson python3-faiss
+    apt install -y python3 python3-dev pipx gcc g++ linux-headers-$(uname -r) libffi-dev cargo cmake autoconf swig ninja-build patchelf meson python3-faiss
 RUN adduser -u 1000 --system --shell /bin/sh --home /home/jovyan jovyan
 USER jovyan
 WORKDIR /home/jovyan