diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
index be6f58b4..d350f5f7 100644
--- a/.github/workflows/main.yml
+++ b/.github/workflows/main.yml
@@ -34,7 +34,8 @@ jobs:
           BUILDS_PATH: builds
           BUILD_METHOD: ""
         with:
-          args: IMAGE_VERSION=${{ matrix.version }} IMAGE_TARGET=${{ matrix.target }}
+          image_version: ${{ matrix.version }}
+          image_target: ${{ matrix.target }}
 
 
       # Upload distributables
diff --git a/Dockerfile b/Dockerfile
index 577e3089..0a8641a7 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,7 +1,7 @@
 ARG IMAGE_VERSION=2019.0.00f0
 ARG IMAGE_TARGET=webgl
 
-FROM gableroux/unity3d:${IMAGE_VERSION}-${IMAGE_TARGET}
+FROM gableroux/unity3d:${INPUT_IMAGE_VERSION}-${INPUT_IMAGE_TARGET}
 
 LABEL "com.github.actions.name"="Unity - Builder"
 LABEL "com.github.actions.description"="Build Unity projects for different platforms."