diff --git a/dags/connection-check.py b/dags/connection-check.py
index e5d4a63..520a11e 100644
--- a/dags/connection-check.py
+++ b/dags/connection-check.py
@@ -2,7 +2,7 @@ from airflow import DAG
 from airflow.decorators import task
 import pendulum
 
-TARGET_HOST = "airflow-k8s-demo-0164-worker-gpu-0.airflow-k8s-demo-0164-worker-gpu"
+TARGET_HOST = "airflow-k8s-demo-0168-worker-gpu-0.airflow-k8s-demo-0168-worker-gpu"
 
 with DAG(
     dag_id='debug_gpu_network_check',
diff --git a/dags/test-train-pytorch.py b/dags/test-train-pytorch.py
index 94ba434..48e219e 100644
--- a/dags/test-train-pytorch.py
+++ b/dags/test-train-pytorch.py
@@ -7,7 +7,7 @@ import json
 
 # --- CONFIGURATION ---
 WORLD_SIZE = 2
-MASTER_ADDR = "airflow-k8s-demo-0167-worker-gpu-0.airflow-k8s-demo-0167-worker-gpu"
+MASTER_ADDR = "airflow-k8s-demo-0168-worker-gpu-0.airflow-k8s-demo-0168-worker-gpu"
 MASTER_PORT = "29500"
 NCCL_TIMEOUT = 1800