diff --git a/Dockerfile b/Dockerfile
index ef2ea32e..6caacd49 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -3,7 +3,7 @@ ARG version=v0.0
 
 FROM eclipse-temurin:17 AS compile
 LABEL maintainer="roan@roanh.dev"
-ARG ref
+ARG version
 WORKDIR /gMark
 ADD gMark/gradle/wrapper/ /gMark/gradle/wrapper/
 ADD gMark/src/ /gMark/src/
@@ -16,7 +16,7 @@ RUN ./gradlew -PrefName=v$version cli:shadowJar
 
 FROM eclipse-temurin:17
 LABEL maintainer="roan@roanh.dev"
-ARG ref
+ARG version
 WORKDIR /gMark
 COPY --from=compile /gMark/cli/build/libs/gMark-v$version.jar ./gMark.jar
 ENTRYPOINT ["java", "-jar", "gMark.jar"]
\ No newline at end of file