diff --git a/CodeGen/tests/test_compose_on_gaudi.sh b/CodeGen/tests/test_compose_on_gaudi.sh
index 493cd0507..8ed2916f3 100644
--- a/CodeGen/tests/test_compose_on_gaudi.sh
+++ b/CodeGen/tests/test_compose_on_gaudi.sh
@@ -196,8 +196,9 @@ function validate_gradio() {
 
 function stop_docker() {
     local docker_profile="$1"
-
-    cd $WORKPATH/docker_compose/intel/hpu/gaudi
+    cd $WORKPATH/docker_compose
+    source set_env.sh
+    cd intel/hpu/gaudi
     docker compose --profile ${docker_profile} down
 }
 
diff --git a/CodeGen/tests/test_compose_on_xeon.sh b/CodeGen/tests/test_compose_on_xeon.sh
index 597521a12..850bb6871 100644
--- a/CodeGen/tests/test_compose_on_xeon.sh
+++ b/CodeGen/tests/test_compose_on_xeon.sh
@@ -199,7 +199,9 @@ function validate_gradio() {
 function stop_docker() {
     local docker_profile="$1"
 
-    cd $WORKPATH/docker_compose/intel/cpu/xeon/
+    cd $WORKPATH/docker_compose
+    source set_env.sh
+    cd intel/cpu/xeon/
     docker compose --profile ${docker_profile} down
 }