diff --git a/Dockerfile b/Dockerfile
index 5f54929..af84aea 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -22,4 +22,4 @@ ENV OPENAI_API_KEY=${OPENAI_API_KEY}
 ENV HUGGINGFACE_API_KEY=${HUGGINGFACE_API_KEY}
 
 # Command to start the Gradio app
-CMD ["python", "src/frontend/archive_gui.py"]
+CMD ["python", "frontend/archive_gui.py"]
diff --git a/infra/Dockerfile b/infra/Dockerfile
index 91a868e..a0c2198 100644
--- a/infra/Dockerfile
+++ b/infra/Dockerfile
@@ -15,4 +15,4 @@ RUN pip install -r requirements.txt
 EXPOSE 5000
 
 # Run the application
-CMD ["python", "src/frontend/gui_chat_interface.py"]
+CMD ["python", "frontend/archive_gui.py"]