diff --git a/Dockerfile b/Dockerfile
index 18b24f0..2e24da9 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -11,6 +11,6 @@ WORKDIR ./app
 
 COPY ./get_dataset.py ./
 COPY ./get_stats.py ./
-COPY ./train.py ./
+COPY train.py ./
 COPY predict.py ./
 CMD bash
\ No newline at end of file
diff --git a/training/train.py b/train.py
similarity index 100%
rename from training/train.py
rename to train.py