diff --git a/Dockerfile b/Dockerfile
index 8be2e1d..c1b2b51 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,10 +1,11 @@
 FROM ubuntu:latest
 
+WORKDIR /subgen
+
 RUN apt-get update && apt-get -y install python3 python3-pip
 
 ADD https://raw.githubusercontent.com/McCloudS/subgen/main/subgen/subgen.py /subgen/subgen.py
 
-ENTRYPOINT ["python3"]
-CMD ["/subgen/subgen.py"]
+CMD ["python", "-u", "subgen.py"]
 
 EXPOSE 8090