diff --git a/Dockerfile b/Dockerfile
index 8456fcf..9d595fd 100755
--- a/Dockerfile
+++ b/Dockerfile
@@ -24,7 +24,7 @@ RUN apt-get update \
     accelerate \
     optimum
 
-ADD https://raw.githubusercontent.com/McCloudS/subgen/transformers/subgen/subgen.py /subgen/subgen.py
+ADD https://raw.githubusercontent.com/McCloudS/subgen/main/subgen/subgen.py /subgen/subgen.py
 
 CMD [ "python3", "-u", "./subgen.py" ]