diff --git a/python/Dockerfile b/python/Dockerfile
index 0bbc90b..a01aa13 100644
--- a/python/Dockerfile
+++ b/python/Dockerfile
@@ -1,5 +1,5 @@
-ARG VERSION="latest"
-FROM registry.tomastm.com/base-debian:${VERSION}
+ARG VERSION="latest-arm64"
+FROM registry.tomastm.com/base-ubuntu:${VERSION}
 
 COPY install-packages.sh /tmp/install-packages.sh
 RUN sudo chmod +x /tmp/install-packages.sh && /tmp/install-packages.sh && sudo rm /tmp/install-packages.sh