diff --git a/Dockerfile b/Dockerfile
index 49b8b4f..3bcfdde 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,5 +1,5 @@
 FROM python:3-slim AS builder
-ADD . /app
+ADD main.py /app/main.py
 WORKDIR /app
 
 RUN pip install --target=/app requests
diff --git a/action.yml b/action.yml
index a6cc022..3d458dc 100644
--- a/action.yml
+++ b/action.yml
@@ -3,7 +3,7 @@ description: "Converts IDE TODO comments to GitHub issues"
 author: "Alastair Mooney"
 runs:
   using: "docker"
-  image: "Dockerfile"
+  image: "docker://ghcr.io/alstr/todo-to-issue-action:v4.4.1"
 branding:
   icon: "check-square"
   color: "orange"