diff --git a/docker/Dockerfile b/docker/Dockerfile
index 1fbb04dd0cb4647fba76268f1da08401b9983dc8..7334ecaeb8ae8464212c44937a5d37ff852b531f 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -61,4 +61,4 @@ ENV XLA_FLAGS="--xla_gpu_enable_triton_gemm=false"
 ENV XLA_PYTHON_CLIENT_PREALLOCATE=true
 ENV XLA_CLIENT_MEM_FRACTION=0.95
 
-CMD ["python3", "run_alphafold.py"]
\ No newline at end of file
+ENTRYPOINT ["python3", "run_alphafold.py"]
\ No newline at end of file