diff --git a/Dockerfile b/Dockerfile
index c8da49d..8cad475 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,3 +1,4 @@
+ARG VERSION=17
 FROM golang AS builder
 
 WORKDIR /app
@@ -9,8 +10,8 @@ COPY ./go.sum /app
 RUN go build -o /app/main /app/src/main.go 
 RUN go build -o /app/cmd /app/src/cmd.go 
 
-ARG JAVA_VERSION=16
-FROM openjdk:16
+
+FROM openjdk:$VERSION
 
 ENV MIN_MEM=512M
 ENV MAX_MEM=1G