diff --git a/Dockerfile b/Dockerfile
index d2f70a5..3124f9f 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -18,4 +18,4 @@ WORKDIR /app/service
 EXPOSE 5000
 EXPOSE 8080
 
-CMD ["python3", "src/run_service.py"]
+CMD ["python3", "src/serve.py"]
diff --git a/setup/docker.sh b/setup/docker.sh
index 93af6fd..f23d4aa 100755
--- a/setup/docker.sh
+++ b/setup/docker.sh
@@ -11,5 +11,5 @@ dvc pull
 
 git submodule update --init --recursive
 
-docker build -f Dockerfile-base -t detr-server-base .
+docker build -f Dockerfile_base -t detr-server-base .
 docker build -f Dockerfile -t detr-server .