diff --git a/publish/action.yml b/publish/action.yml
index a7bcee0..6246318 100644
--- a/publish/action.yml
+++ b/publish/action.yml
@@ -1,6 +1,11 @@
 name: "Publish Library"
 description: "Publish gradle artifact"
 
+inputs:
+  version:
+    description: "Library Version"
+    required: true
+
 runs:
   using: "composite"
   steps:
@@ -15,4 +20,4 @@ runs:
         BINOM_REPO_URL: ${{ vars.BINOM_REPO_URL }}
         BINOM_REPO_USER: ${{ vars.BINOM_REPO_USER }}
       shell: bash
-      run: "./gradlew -Pversion=${{ inputs.image_name }} publish -Dorg.gradle.jvmargs=-Xmx4096M --parallel --no-daemon --no-watch-fs --stacktrace"
+      run: "./gradlew -Pversion=${{ inputs.version }} publish -Dorg.gradle.jvmargs=-Xmx4096M --parallel --no-daemon --no-watch-fs --stacktrace"