diff --git a/Dockerfile b/Dockerfile
index b2e85c5..af80ad2 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -27,8 +27,8 @@ COPY . .
 
 RUN pip install -r requirements.txt
 
-WORKDIR src
+ENV PYTHONPATH "${PYTHONPATH}:src"
 
-ENTRYPOINT ["python", "main.py"]
+ENTRYPOINT ["python", "src/main.py"]