From 92e5a6638228d286a540c71f7330e8ea839c3542 Mon Sep 17 00:00:00 2001
From: Alexandre <44178713+alexbelgium@users.noreply.github.com>
Date: Sun, 21 May 2023 10:27:29 +0200
Subject: [PATCH] Update Dockerfile

---
 portainer_agent/Dockerfile | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/portainer_agent/Dockerfile b/portainer_agent/Dockerfile
index 4ad673a04..df9e17f51 100644
--- a/portainer_agent/Dockerfile
+++ b/portainer_agent/Dockerfile
@@ -14,12 +14,14 @@
 # 1 Build Image #
 #################
 
+ARG BUILD_FROM
+FROM $BUILD_FROM as base_image
+
 FROM portainer/agent:alpine as original_agent
 ENV PORTAINER_AGENT_ARGS=""
 
-ARG BUILD_FROM
-ENV BUILD_BASE="$BUILD_FROM"
-FROM $BUILD_BASE
+# Get initial base
+COPY --from=base_image / /
 
 # Copy Portainer agent binaries
 COPY --from=original_agent /app /app