diff --git a/arch/docker-compose.yaml b/arch/docker-compose.yaml
index 11424166..dc832b4f 100644
--- a/arch/docker-compose.yaml
+++ b/arch/docker-compose.yaml
@@ -12,5 +12,3 @@ services:
       - ~/archgw_logs:/var/log/
     extra_hosts:
       - "host.docker.internal:host-gateway"
-    env_file:
-      - stage.env
diff --git a/e2e_tests/run_e2e_tests.sh b/e2e_tests/run_e2e_tests.sh
index ea0c65d9..44845b90 100644
--- a/e2e_tests/run_e2e_tests.sh
+++ b/e2e_tests/run_e2e_tests.sh
@@ -6,7 +6,18 @@ pwd
 
 . ./common_scripts.sh
 
+log building llm and prompt gateway rust modules
+log ============================================
+cd ../arch
+sh build_filter_image.sh
+log starting the arch gateway service
+docker compose -f docker-compose.yaml down
+docker compose -f docker-compose.yaml up -d
+wait_for_healthz "http://localhost:10000/healthz" 60
+cd -
+
 log building function_callling demo
+log ===============================
 cd ../demos/function_calling
 docker compose build
 
@@ -15,32 +26,26 @@ docker compose up -d
 cd -
 
 log building model server
+log =====================
 cd ../model_server
 poetry install
 log starting model server
 poetry run archgw_modelserver restart
 cd -
 
-log building llm and prompt gateway rust modules
-cd ../arch
-sh build_filter_image.sh
-log starting the arch gateway service
-docker compose -f docker-compose.yaml down
-docker compose -f docker-compose.yaml up -d
-cd -
-
-wait_for_healthz "http://localhost:10000/healthz" 60
-
 log running e2e tests
+log =================
 poetry install
 poetry run pytest
 
 log shutting down the arch gateway service
+log ======================================
 cd ../arch
 docker compose -f docker-compose.yaml stop
 cd -
 
 log shutting down the function_calling demo
+log =======================================
 cd ../demos/function_calling
 docker compose down
 cd -