diff --git a/lab5/train/Dockerfile b/lab5/train/Dockerfile
index f3598ed..5e579a5 100644
--- a/lab5/train/Dockerfile
+++ b/lab5/train/Dockerfile
@@ -8,7 +8,7 @@ RUN apt install -y unzip >>/dev/null
 
 WORKDIR /app
 
-COPY ./test_eval.py ./
+COPY ./train.py ./
 COPY ./script.sh ./
 RUN chmod +x script.sh