diff --git a/dockerfiles/python/Dockerfile b/dockerfiles/python/Dockerfile
index 4d6ac18..d49a829 100644
--- a/dockerfiles/python/Dockerfile
+++ b/dockerfiles/python/Dockerfile
@@ -1,7 +1,6 @@
-FROM icfpcontest2020/python:latest
-WORKDIR /source
+FROM icfpcontest2020/python
 
+WORKDIR /solution
 COPY . .
-WORKDIR /source/app
-
-ENTRYPOINT ["python", "main.py"]
+RUN ./build.sh
+ENTRYPOINT ["./run.sh"]