diff --git a/.github/workflows/pr-test.yml b/.github/workflows/pr-test.yml
index 9b671a0e..8db4b250 100644
--- a/.github/workflows/pr-test.yml
+++ b/.github/workflows/pr-test.yml
@@ -1139,6 +1139,7 @@ jobs:
           docker run -d --name tai-k8s \
             --network k3d-tai-test \
             -p 8081:8080 -p 9101:9100 -p 6443:6443 -p 6081:6080 \
+            -v /var/run/docker.sock:/var/run/docker.sock:ro \
             -v /tmp/kubeconfig-tai-k8s.yml:/etc/tai/kubeconfig.yml:ro \
             -e TAI_K8S_UPSTREAM="tcp://${K3D_IP}:6443" \
             -e TAI_KUBECONFIG=/etc/tai/kubeconfig.yml \
@@ -1981,7 +1982,10 @@ jobs:
           docker run -d --name tai-k8s \
             --network k3d-tai-test \
             -p 8081:8080 -p 9101:9100 -p 6443:6443 -p 6081:6080 \
+            -v /var/run/docker.sock:/var/run/docker.sock:ro \
+            -v /tmp/kubeconfig-tai-k8s.yml:/etc/tai/kubeconfig.yml:ro \
             -e TAI_K8S_UPSTREAM="tcp://${K3D_IP}:6443" \
+            -e TAI_KUBECONFIG=/etc/tai/kubeconfig.yml \
             yaoapp/tai:latest
 
           for i in $(seq 1 30); do
@@ -2006,7 +2010,10 @@ jobs:
 
       - name: Generate kubeconfig for benchmarks
         run: |
+          K3D_IP=$(docker inspect k3d-tai-test-server-0 | jq -r '.[0].NetworkSettings.Networks["k3d-tai-test"].IPAddress')
           k3d kubeconfig get tai-test > /tmp/kubeconfig-k3d.yml
+          sed "s|server: .*|server: https://${K3D_IP}:6443|" /tmp/kubeconfig-k3d.yml \
+            > /tmp/kubeconfig-tai-k8s.yml
           sed 's|server: .*|server: https://127.0.0.1:6443|' /tmp/kubeconfig-k3d.yml \
             > ${{ runner.temp }}/kubeconfig-tai.yml
 
diff --git a/.github/workflows/unit-test.yml b/.github/workflows/unit-test.yml
index 2173f606..31920fc1 100644
--- a/.github/workflows/unit-test.yml
+++ b/.github/workflows/unit-test.yml
@@ -847,6 +847,7 @@ jobs:
           docker run -d --name tai-k8s \
             --network k3d-tai-test \
             -p 8081:8080 -p 9101:9100 -p 6443:6443 -p 6081:6080 \
+            -v /var/run/docker.sock:/var/run/docker.sock:ro \
             -v /tmp/kubeconfig-tai-k8s.yml:/etc/tai/kubeconfig.yml:ro \
             -e TAI_K8S_UPSTREAM="tcp://${K3D_IP}:6443" \
             -e TAI_KUBECONFIG=/etc/tai/kubeconfig.yml \
@@ -1487,7 +1488,10 @@ jobs:
           docker run -d --name tai-k8s \
             --network k3d-tai-test \
             -p 8081:8080 -p 9101:9100 -p 6443:6443 -p 6081:6080 \
+            -v /var/run/docker.sock:/var/run/docker.sock:ro \
+            -v /tmp/kubeconfig-tai-k8s.yml:/etc/tai/kubeconfig.yml:ro \
             -e TAI_K8S_UPSTREAM="tcp://${K3D_IP}:6443" \
+            -e TAI_KUBECONFIG=/etc/tai/kubeconfig.yml \
             yaoapp/tai:latest
 
           for i in $(seq 1 30); do
@@ -1512,7 +1516,10 @@ jobs:
 
       - name: Generate kubeconfig for benchmarks
         run: |
+          K3D_IP=$(docker inspect k3d-tai-test-server-0 | jq -r '.[0].NetworkSettings.Networks["k3d-tai-test"].IPAddress')
           k3d kubeconfig get tai-test > /tmp/kubeconfig-k3d.yml
+          sed "s|server: .*|server: https://${K3D_IP}:6443|" /tmp/kubeconfig-k3d.yml \
+            > /tmp/kubeconfig-tai-k8s.yml
           sed 's|server: .*|server: https://127.0.0.1:6443|' /tmp/kubeconfig-k3d.yml \
             > ${{ runner.temp }}/kubeconfig-tai.yml