diff --git a/docker/docker-compose.yml b/docker/docker-compose.yml
index 801d78c..2ff82a0 100644
--- a/docker/docker-compose.yml
+++ b/docker/docker-compose.yml
@@ -19,7 +19,7 @@ services:
        - db
     image: sartography/protocol-builder-mock:$E2E_TAG
     environment:
-      - FLASK_APP=/protocol-builder-mock/app.py
+      - FLASK_APP=./pb/__init__.py
       - UPGRADE_DB=true
       - DB_HOST=db
       - DB_PORT=5432