diff --git a/portainer_agent/Dockerfile b/portainer_agent/Dockerfile
index 2e29aa2ef..f11e6940d 100644
--- a/portainer_agent/Dockerfile
+++ b/portainer_agent/Dockerfile
@@ -14,10 +14,11 @@
 # 1 Build Image #
 #################
 
+ARG BUILD_FROM
 FROM portainer/agent:alpine as original_agent
 ENV PORTAINER_AGENT_ARGS=""
 
-FROM ghcr.io/linuxserver/baseimage-alpine:latest
+FROM $BUILD_FROM
 
 # Copy Portainer agent binaries
 COPY --from=original_agent /app /app