diff --git a/Dockerfile b/Dockerfile
deleted file mode 100644
index 8533c80..0000000
--- a/Dockerfile
+++ /dev/null
@@ -1,7 +0,0 @@
-FROM authelia/authelia:latest
-
-LABEL org.opencontainers.image.authors="Nicholas Malcolm"
-LABEL org.opencontainers.image.source="https://github.com/bubylou/authelia-validate"
-
-COPY entrypoint.sh /entrypoint.sh
-ENTRYPOINT ["/entrypoint.sh"]
diff --git a/README.md b/README.md
index d4af6a8..937c913 100644
--- a/README.md
+++ b/README.md
@@ -8,6 +8,10 @@ This action run a validation check on an authelia configuration file. The output
 
 **Required** The file path from the repositroy root to the configuration.yml file. Default `./configuration.yml`.
 
+### `version`
+
+**Required** Authelia version to validate the configuration file against. Default `latest`.
+
 ## Enironment Variables
 
 If the configuration file excludes any setting in favor of pulling them from an environment variable then those variables will also need to be set in the action. Any string can be used instead of actual enviroment variables being used with the exception of `AUTHELIA_STORAGE_ENCRYPTION_KEY` which must be at least 20 characters long. An example of using these dummy environment variables is included in the example below.
@@ -21,11 +25,12 @@ steps:
     uses: actions/checkout@v4
 
   - name: Validate config
-    uses: bubylou/authelia-validate@v2
+    uses: bubylou/authelia-validate@v3
     env:
       AUTHELIA_JWT_SECRET: test
       AUTHELIA_SESSION_SECRET: test
       AUTHELIA_STORAGE_ENCRYPTION_KEY: twentycharacterslong
     with:
       file-path: ./authelia/configuration.yml
+      version: v4.37.5
 ```
diff --git a/action.yml b/action.yml
index 6638463..62734f6 100644
--- a/action.yml
+++ b/action.yml
@@ -9,8 +9,13 @@ inputs:
     description: 'Path to configuration file'
     required: true
     default: './configuration.yml'
+  version:
+    description: 'Version of Authelia to check against'
+    required: true
+    default: 'latest'
 runs:
   using: 'docker'
-  image: 'Dockerfile'
+  image: authelia:${{ inputs.version }}
+  entrypoint: 'entrypoint.sh'
   args:
     - ${{ inputs.file-path }}