diff --git a/Dockerfile b/Dockerfile
index a7a4d02..1ec3209 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,5 +1,6 @@
 # Build stage
 ARG MAVEN_IMAGE=maven:3.9.16-eclipse-temurin-17-alpine
+ARG RUNTIME_IMAGE=eclipse-temurin:17-jre-alpine
 FROM ${MAVEN_IMAGE} AS builder
 WORKDIR /app
 COPY .mvn/settings.xml /root/.m2/settings.xml
@@ -9,7 +10,6 @@ COPY src ./src
 RUN mvn -B -ntp -Dmaven.test.skip=true clean package
 
 # Runtime stage
-ARG RUNTIME_IMAGE=eclipse-temurin:17-jre-alpine
 FROM ${RUNTIME_IMAGE}
 WORKDIR /app
 COPY --from=builder /app/target/agent-foundation-0.0.1-SNAPSHOT.jar app.jar