diff --git a/Dockerfile b/Dockerfile
index ec50333..4c15aa7 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -65,4 +65,4 @@ RUN rm -r *.zip
 RUN wget "https://github.com/andreped/neukit/releases/download/test-data/test_thorax_CT.nii.gz"
 
 # CMD ["/bin/bash"]
-CMD ["python3", "demo/app.py"]
+CMD ["python3", "app.py"]
diff --git a/README.md b/README.md
index 803bd49..e00ead9 100644
--- a/README.md
+++ b/README.md
@@ -7,7 +7,7 @@ app_port: 7860
 emoji: 🧠
 pinned: false
 license: mit
-app_file: demo/app.py
+app_file: app.py
 ---
 
 <div align="center">