diff --git a/machine-learning/Dockerfile b/machine-learning/Dockerfile
index d6b45d21d..114c0ad91 100644
--- a/machine-learning/Dockerfile
+++ b/machine-learning/Dockerfile
@@ -25,4 +25,4 @@ WORKDIR /usr/src/app
 
 COPY . .
 ENV PYTHONPATH=`pwd`
-CMD ["python", "main.py"]
\ No newline at end of file
+CMD ["python", "src/main.py"]