diff --git a/.github/workflows/train.yml b/.github/workflows/train.yml
index a5d1514..4a0d3db 100644
--- a/.github/workflows/train.yml
+++ b/.github/workflows/train.yml
@@ -15,7 +15,7 @@ jobs:
 
       - name: Use Docker action
         id: train
-        uses: ./
+        uses: ./train-action/
         with:
           epochs: 1
 
diff --git a/train.dockerfile b/train-action/Dockerfile
similarity index 90%
rename from train.dockerfile
rename to train-action/Dockerfile
index 28441db..463c862 100644
--- a/train.dockerfile
+++ b/train-action/Dockerfile
@@ -14,4 +14,4 @@ RUN wget https://bendun.students.wmi.amu.edu.pl/stop_times.categories.tsv
 
 VOLUME /github/workspace/
 
-ENTRYPOINT ["/ium/train-entrypoint.sh"]
+ENTRYPOINT ["/ium/train-action/entrypoint.sh"]
diff --git a/train-entrypoint.sh b/train-action/entrypoint.sh
similarity index 100%
rename from train-entrypoint.sh
rename to train-action/entrypoint.sh
diff --git a/train.yml b/train-action/train.yml
similarity index 87%
rename from train.yml
rename to train-action/train.yml
index 329be13..597e32f 100644
--- a/train.yml
+++ b/train-action/train.yml
@@ -7,6 +7,6 @@ inputs:
     default: '2'
 runs:
   using: 'docker'
-  image: 'train.dockerfile'
+  image: 'Dockerfile'
   args:
     - ${{ inputs.epochs }}