diff --git a/Dockerfile b/Dockerfile
index 4d1c31f..8279369 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,9 +1,7 @@
 FROM python:3.6.12-alpine3.12
 
-ENV APP_PATH=app
-
 RUN apk add --no-cache git
 RUN python -m pip install --no-cache-dir --upgrade pip
 RUN pip install --no-cache-dir git+git://github.com/Falldog/django-migration-checker
 
-ENTRYPOINT django-find-conflicts $APP_PATH
+ENTRYPOINT ["django-find-conflicts"]
diff --git a/action.yml b/action.yml
index 51608ed..447ef56 100644
--- a/action.yml
+++ b/action.yml
@@ -12,6 +12,6 @@ inputs:
 
 runs:
   using: docker
-  image: Dockerfile
-  env: 
-    APP_PATH: ${{ inputs.app-path }}
+  image: hardcoretech/django-migration-checker:latest
+  args: 
+    - ${{ inputs.app-path }}