diff --git a/Dockerfile b/Dockerfile
index 662c543..9896245 100755
--- a/Dockerfile
+++ b/Dockerfile
@@ -8,6 +8,6 @@ ENV PYTHONUNBUFFERED 1
 
 ADD https://raw.githubusercontent.com/McCloudS/subgen/main/subgen/subgen.py /subgen/subgen.py
 
-CMD [ "python3", "-u", "subgen/subgen.py" ]
+CMD [ "python3", "-u", "./subgen.py" ]
 
 EXPOSE 8090
diff --git a/Dockerfile.cuda b/Dockerfile.cuda
index db7d4e1..3d3b26a 100644
--- a/Dockerfile.cuda
+++ b/Dockerfile.cuda
@@ -8,6 +8,6 @@ ENV PYTHONUNBUFFERED 1
 
 ADD https://raw.githubusercontent.com/McCloudS/subgen/main/subgen/subgen.py /subgen/subgen.py
 
-CMD [ "python3", "-u", "subgen/subgen.py" ]
+CMD [ "python3", "-u", "./subgen.py" ]
 
 EXPOSE 8090